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

Question

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

3

我按照基础教程https://www.tensorflow.org/text/tutorials/text_generation#train_the_model生成了一个基本的RNN模型，我逐行遵循教程操作（只更改了模型参数和数据集）。当训练进行到3/4的时候我停止了。现在模型已经生成了14个检查点文件和14个检查点索引文件。

我尝试使用model.load_weights(os.path.join(checkpoint_dir, "checkpoint"))加载检查点，但是出现了以下错误：OSError: Unable to open file (file signature not found)，通过查询网络我发现这是因为文件未按要求编码所致。但是主检查点文件没有文件扩展名也没有编码。

请问我做错了什么吗？

- Fabrizio

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Jérémy · Accepted Answer

我曾经在手动保存权重时使用了model.save_weights("weights")，然后遇到了相同的错误。对我来说，解决方法是改用：model.load_weights("weights")，而不是：model.load_weights("checkpoint")。因此，错误在于我在model.load_weights()函数中使用了错误的名称。我猜你也有同样的问题，你的model.load_weights()中的"checkpoint"需要替换为不同的名称。