AlexeyAB
2017-09-11 99fc4ace9034e95549a068cda6b3892234bb0f7d
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
}