我遵循https://www.tensorflow.org/text/tutorials/text_generation#train_the_model的基本教程,这导致基本RNN模型的生成。我一行一行地遵循教程(仅更改模型参数和数据集)。训练进行到四分之三的时候我就停止了。生成的模型已经14检查点文件和14个检查站指标文件。
我试图加载检查点使用model.load_weights(os.path.join(checkpoint_dir, "checkpoint"))
但我得到以下错误OSError: Unable to open file (file signature not found)
,,通过网络我发现它指编码的文件不是所需的编码。但是主检查点文件没有文件扩展名,也没有编码。
我做错了什么吗?
在手动使用model.save_weights("weights")
保存权重后,我遇到了同样的错误。我的解决方案是使用:
model.load_weights("weights")
代替:
model.load_weights("checkpoint")
所以错误是我在model.load_weights()
函数中使用了错误的名称。我猜你也有同样的问题,你的model.load_weights()
中的"checkpoint"
必须用不同的名称代替。