TensorFlow，训练产生3个ckpt文件，哪个是正确的？(Music_RNN_RBM)

Question

我在给Music_RNN_RBM一个机会

我已经训练了模型，并且，正如说明所说的那样，生成了一个ckpt文件，问题是在保存的每个时期的目录中，我有三个.ckpt .data .index和.meta示例：

model.ckpt-151800.data-00000-of-00001
model.ckpt-151800.index
model.ckpt-151800.meta

现在，下一步将是生成低谷

python rnn_rbm_generate.py <path_to_ckpt_file>

对于每种类型的文件，我都尝试过这样做

python3 rnn_rbm_generate.py model.ckpt-151800.index

但是我每次都会得到这个错误

Traceback (most recent call last):
  File "rnn_rbm_generate.py", line 52, in <module>
    main(sys.argv[1])
  File "rnn_rbm_generate.py", line 43, in main
    saver.restore(sess, ckpt.model_checkpoint_path)
AttributeError: 'NoneType' object has no attribute 'model_checkpoint_path' 

我不知道我该做什么

Answer 1

正如您所发现的，每个检查点都由.data、.meta和index文件组成。

但是，TensorFlow还会创建一个checkpoints文件，用于跟踪输出目录中的所有检查点。它包含一个指向最新检查点的链接model_checkpoint_path，这就是您的模型正在寻找的。

因此，为了回答您的问题，您提供的路径应该包含此checkpoints文件。

Answer 2

有一个function to load the latest one。使用它来加载它，并创建一个适当的保存模型，检查点被用作备份，或者在之后继续训练，而不是服务。前面提到的“检查点”文件是由我提到的函数加载的，它是关于正在创建的文件的信息，无法加载。问题，问吧

编辑:除了存储库中的代码外，您还可以提供完整的文件夹，而不是文件。