mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
🔥 🔥 MEMORY LEAK!!! 🔥 🔥
This commit is contained in:
parent
59ed1719d4
commit
248d9f179d
@ -526,12 +526,6 @@ void forward_region_layer_gpu(const layer l, network net)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
float *truth_cpu = 0;
|
|
||||||
if(net.truth_gpu){
|
|
||||||
int num_truth = l.batch*l.truths;
|
|
||||||
truth_cpu = calloc(num_truth, sizeof(float));
|
|
||||||
cuda_pull_array(net.truth_gpu, truth_cpu, num_truth);
|
|
||||||
}
|
|
||||||
cuda_pull_array(l.output_gpu, net.input, l.batch*l.inputs);
|
cuda_pull_array(l.output_gpu, net.input, l.batch*l.inputs);
|
||||||
forward_region_layer(l, net);
|
forward_region_layer(l, net);
|
||||||
//cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs);
|
//cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user