OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed memory release

This commit is contained in:
AlexeyAB 2017-07-25 14:25:36 +03:00
parent 51477ab274
commit 576dbe12e6
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/network.c
View File

@ -595,11 +595,14 @@ void free_network(network net)
free_layer(net.layers[i]); free_layer(net.layers[i]);
} }
free(net.layers); free(net.layers);
free(net.workspace);
#ifdef GPU #ifdef GPU
if (gpu_index >= 0) cuda_free(net.workspace);
else free(net.workspace);
if (*net.input_gpu) cuda_free(*net.input_gpu); if (*net.input_gpu) cuda_free(*net.input_gpu);
if (*net.truth_gpu) cuda_free(*net.truth_gpu); if (*net.truth_gpu) cuda_free(*net.truth_gpu);
if (net.input_gpu) free(net.input_gpu); if (net.input_gpu) free(net.input_gpu);
if (net.truth_gpu) free(net.truth_gpu); if (net.truth_gpu) free(net.truth_gpu);
#else
free(net.workspace);
#endif #endif
} }