AlexeyAB
2017-08-17 4d2fefd75a57dfd6e60680eaf7408c82e15a025d
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
}