这篇教程用keras.backend.clear_session()方法可以有效解决模型的内存占用问题。写得很实用，希望能帮到您。
在用tensorflow运行keras模型时，我发现运行完model.predict()之后，模型并不会自动关闭，还是回留存在RAM中，如果使用 for loop循环使用model.predict时，就会发生内存崩溃的情况。
用keras.backend.clear_session()方法可以有效解决模型的内存占用问题。
from tensorflow.keras.backend import clear_session
from tensorflow.keras.models import load_model
for i in range(number):
clear_session()
model = load_model(model_path)
————————————————
版权声明：本文为CSDN博主「Cmmm丶」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_46090057/article/details/109219082
返回列表
16型人格分析（MBTI） |
