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

我正在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'

我不知道该怎么办

最佳答案

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

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

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