mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Added COCO, fixed memory leaks
This commit is contained in:
@ -204,6 +204,7 @@ void backward_detection_layer_gpu(detection_layer l, network_state state)
|
||||
backward_detection_layer(l, cpu_state);
|
||||
cuda_push_array(state.delta, delta_cpu, l.batch*l.inputs);
|
||||
|
||||
if (truth_cpu) free(truth_cpu);
|
||||
free(in_cpu);
|
||||
free(delta_cpu);
|
||||
}
|
||||
|
Reference in New Issue
Block a user