AlexeyAB
2018-01-25 8b94d474d591dd76d146c9aeed029d0b5836ac3f
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);
}