AlexeyAB
2018-01-17 54f83e153549dd1a63bcc8fa5e55fb171621a989
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);
}