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。使用它来加载它，并创建正确的保存模型，将检查点用作备份，或者在以后继续训练而不使用。我提到的功能会加载提到的“ checkpoints”文件，它是有关正在创建的文件的信息，无法加载。问题，请问

编辑： 回购中的代码除了表明您还可以提供完整文件夹而不是文件之外。