Minor fix

This commit is contained in:
AlexeyAB
2018-10-28 23:58:08 +03:00
parent 31df5e2735
commit 08e8e0c8c2
2 changed files with 2 additions and 1 deletions

View File

@ -54,6 +54,7 @@ void free_layer(layer l)
if (l.r_cpu) free(l.r_cpu);
if (l.h_cpu) free(l.h_cpu);
if (l.binary_input) free(l.binary_input);
if (l.loss) free(l.loss);
#ifdef GPU
if (l.indexes_gpu) cuda_free((float *)l.indexes_gpu);