我正在使用Attention机制进行图像字幕处理，并且我手动保存了所有图层的权重，但是当我重新启动我的PC并加载保存的权重时，模型的损失增加了太多，看来权重没有得到正确的保存。 但我没有发现任何未保存的重量。有人可以帮忙吗？
答案 0 :(得分：1)
保存权重就可以了。保存权重不会保留其他必须退出的信息，例如退出优化程序后必须恢复的信息。因此，在程序中有一条语句，可以在结束执行之前使用model.save保存整个模型。然后，当您重新启动程序时，使用model.load重新加载整个模型。文档为here.