AlexeyAB
2018-02-07 4373f897f1a5efd1a58746c44430dc33df925415
src/region_layer.c
@@ -419,6 +419,7 @@
    free(cpu_state.input);
    if(!state.train) return;
    cuda_push_array(l.delta_gpu, l.delta, l.batch*l.outputs);
   cudaStreamSynchronize(get_cuda_stream());
    if(cpu_state.truth) free(cpu_state.truth);
}