AlexeyAB
2017-07-25 576dbe12e64697f10aaa15f6ef23e0a3d5eea5b5
Fixed memory release
1 files modified
5 ■■■■ changed files
src/network.c 5 ●●●● patch | view | raw | blame | history
src/network.c
@@ -595,11 +595,14 @@
        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
}