无法在 Tensorflow 中加载检查点文件。(OSError:找不到文件签名)



我遵循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"必须用不同的名称代替。

最新更新