问Tensorflow模型权重未完全保存
EN

Stack Overflow用户

提问于 2020-10-08 21:15:34

回答 1查看 124关注 0票数 0

我正在使用图像字幕的注意机制，我手动保存了所有图层的权重，但当我重新启动pc并加载保存的权重时，模型的损失增加了太多，看起来权重没有正确保存。但我没有发现任何未保存的重量。有人能帮上忙吗？

发布于 2020-10-08 23:22:45

只要你在运行你的程序，保存权重是很好的。保存权重不会保留退出执行后必须恢复的其他所需信息，例如优化器的状态。因此，在您的程序中，有一条语句可以在结束执行之前使用model.save保存整个模型。然后，当您重新启动程序时，使用model.load重新加载整个模型。文档是here.

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/64263348

复制

相似问题

问Tensorflow模型权重未完全保存EN