Fix memory leak due to repeatedly calling model.predict(...)

compomics / DeepLC

DeepLC: Retention time prediction for (modified) peptides using Deep Learning.

https://iomics.ugent.be/deeplc

Apache License 2.0

52 stars 18 forks source link

Fix memory leak due to repeatedly calling model.predict(...) #60

Closed markmipt closed 1 year ago

markmipt commented 1 year ago

See this issue for details: https://github.com/keras-team/keras/issues/13118

The changes I've made at least fix that memory leak for me when predictions are split into the batches. Also, I did not noticed any performance drops due to the fix.

Also, the second commit fixes the TF warnings supression.

Regards, Mark