AlexeyAB
2018-01-04 e205c1e7aeb47e3dffd35d1b5ce7841d24b9aff4
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;