Tensorflow模型权重没有完全保存

Question

我使用注意力机制进行图像字幕，我手动保存了所有层的权重，但当我重新启动电脑并加载保存的权重时，模型的损失增加了太多，似乎没有正确保存权重。但我没有发现任何未节省的重量。有人能帮忙吗？

Colab Link

Answer 1

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