Added COCO, fixed memory leaks

This commit is contained in:
Joseph Redmon
2015-07-30 16:19:14 -07:00
parent 98813ad520
commit 846b3b4366
10 changed files with 359 additions and 22 deletions

View File

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