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

Fixed memory release

Browse Source
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
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(net.layers);
free(net.workspace);
#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.truth_gpu) cuda_free(*net.truth_gpu);
if (net.input_gpu) free(net.input_gpu);
if (net.truth_gpu) free(net.truth_gpu);
#else
free(net.workspace);
#endif
}