python - 无法在 Tensorflow 中加载检查点文件。 (操作系统错误 : file signature not found)

Question

我按照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)，通过在线查看，我发现它指的是文件未按所需编码进行编码。但是主检查点文件没有文件扩展名和编码。

我做错了什么吗？

Answer 1

在使用 model.save_weights("weights") 手动保存权重后，我遇到了同样的错误。我的解决方案是使用:

model.load_weights("权重")

代替:

model.load_weights("检查点")

所以错误是因为我在 model.load_weights() 函数中使用了错误的名称。我猜你有同样的问题，你的 model.load_weights() 中的 "checkpoint" 必须用不同的名称替换。