如何解决Tensorflow模型权重并未完全节省
我正在使用Attention机制进行图像字幕处理，并且我手动保存了所有图层的权重，但是当我重新启动我的PC并加载保存的权重时，模型的损失增加了太多，看来权重没有得到正确的保存。 但我没有发现任何未保存的重量。有人可以帮忙吗？
解决方法
只要运行程序，
保存权重就可以了。保存权重不会保留其他必须退出的信息，例如退出优化程序后必须恢复的信息。因此，在程序中有一条语句，可以在结束执行之前使用model.save保存整个模型。然后，当您重新启动程序时，使用model.load重新加载整个模型。文档为here.
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
