AlexeyAB
2017-09-12 6f1ce1f3e8a9aa5f189a7a57e252de5953676809
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
}