AlexeyAB
2018-02-07 f03d486760da03c690f069b23e72fc33bd76f3ea
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);
}