mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Memory leak fixed

Browse Source
This commit is contained in:
AlexeyAB
2017-07-03 14:12:52 +03:00
parent 5eb552fcfc
commit 5bc62b14e0
3 changed files with 97 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/network.c
View File

@ -590,15 +590,16 @@ float network_accuracy_multi(network net, data d, int n)
void free_network(network net)
{
int i;
for(i = 0; i < net.n; ++i){
free_layer(net.layers[i]);
}
free(net.layers);
int i;
for (i = 0; i < net.n; ++i) {
free_layer(net.layers[i]);
}
free(net.layers);
free(net.workspace);
#ifdef GPU
if(*net.input_gpu) cuda_free(*net.input_gpu);
if(*net.truth_gpu) cuda_free(*net.truth_gpu);
if(net.input_gpu) free(net.input_gpu);
if(net.truth_gpu) free(net.truth_gpu);
if (*net.input_gpu) cuda_free(*net.input_gpu);
if (*net.truth_gpu) cuda_free(*net.truth_gpu);
if (net.input_gpu) free(net.input_gpu);
if (net.truth_gpu) free(net.truth_gpu);
#endif
}