Fixed memory leaks for Yolo: train, test

This commit is contained in:
AlexeyAB
2018-05-23 18:27:18 +03:00
parent 768b190b10
commit ec68838342
8 changed files with 73 additions and 13 deletions

View File

@ -11,6 +11,7 @@ void free_layer(layer l)
#endif
return;
}
if (l.mask) free(l.mask);
if (l.cweights) free(l.cweights);
if (l.indexes) free(l.indexes);
if (l.input_layers) free(l.input_layers);