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); }