我正在尝试 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
文件。
关于python - TensorFlow,训练产生3个ckpt文件,哪一个是正确的? (音乐_RNN_RBM),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55071210/