Tensorflow模型权重并未完全节省

问题描述

我正在使用Attention机制进行图像字幕处理,并且我手动保存了所有图层的权重,但是当我重新启动我的PC并加载保存的权重时,模型的损失增加了太多,看来权重没有得到正确的保存。 但我没有发现任何未保存的重量。有人可以帮忙吗?

Colab Link

解决方法

只要运行程序,

保存权重就可以了。保存权重不会保留其他必须退出的信息,例如退出优化程序后必须恢复的信息。因此,在程序中有一条语句,可以在结束执行之前使用model.save保存整个模型。然后,当您重新启动程序时,使用model.load重新加载整个模型。文档为here.

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...