python - TensorFlow，训练产生3个ckpt文件，哪一个是正确的？ (音乐_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 文件。