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 }