ISSUE
https://github.com/tensorflow/tensorflow/issues/14356
解决办法
在每次Infer之后使用keras.backend.clear_session()
import keras
xxxx
keras.backend.clear_session()
本文针对TensorFlow在Infer操作后出现的内存泄漏问题提供了解决方案。通过在每次Infer后调用keras.backend.clear_session(),可以有效释放不再使用的资源,避免内存占用持续增加。
https://github.com/tensorflow/tensorflow/issues/14356
在每次Infer之后使用keras.backend.clear_session()
import keras
xxxx
keras.backend.clear_session()
您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型
6744
1万+
1320
626

被折叠的 条评论
为什么被折叠?