OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Compile fix

This commit is contained in:
AlexeyAB 2020-01-06 23:38:21 +03:00
parent b267e34487
commit c2ce28bbcc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/layer.c
View File

@ -192,9 +192,9 @@ void free_layer_custom(layer l, int keep_cudnn_desc)
if (l.rand_gpu) cuda_free(l.rand_gpu);
if (l.squared_gpu) cuda_free(l.squared_gpu);
if (l.norms_gpu) cuda_free(l.norms_gpu);
if (l.input_sizes_gpu) cuda_free(l.input_sizes_gpu);
if (l.layers_output_gpu) cuda_free(l.layers_output_gpu);
if (l.layers_delta_gpu) cuda_free(l.layers_delta_gpu);
if (l.input_sizes_gpu) cuda_free((float*)l.input_sizes_gpu);
if (l.layers_output_gpu) cuda_free((float*)l.layers_output_gpu);
if (l.layers_delta_gpu) cuda_free((float*)l.layers_delta_gpu);
// CONV-LSTM
if (l.f_gpu) cuda_free(l.f_gpu);