AlexeyAB
2018-01-04 9ac78d8b84f6a059d2cefe22a10aa60de5b3feaf
src/region_layer.c
@@ -409,6 +409,7 @@
        cuda_pull_array(state.truth, truth_cpu, num_truth);
    }
    cuda_pull_array(l.output_gpu, in_cpu, l.batch*l.inputs);
   cudaStreamSynchronize(get_cuda_stream());
    network_state cpu_state = state;
    cpu_state.train = state.train;
    cpu_state.truth = truth_cpu;