将tensorflow更新到最新版本后,我的检查点无法从磁盘加载。我得到一个如下的错误:
...
File "/usr/lib/python2.7/site-packages/tensorflow/python/training/saver.py", line 1560, in restore
{self.saver_def.filename_tensor_name: save_path})
File "/usr/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 895, in run
run_metadata_ptr)
File "/usr/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1124, in _run
feed_dict_tensor, options, run_metadata)
File "/usr/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1321, in _do_run
options, run_metadata)
File "/usr/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1340, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.NotFoundError: Key model/rnn/multi_rnn_cell/cell_0/lstm_cell/kernel not found in checkpoint
[[Node: model/save/RestoreV2_4 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/cpu:0"](_arg_model/save/Const_0_0, model/save/RestoreV2_4/tensor_names, model/save/RestoreV2_4/shape_and_slices)]]
变量名已更改,需要在检查点中重命名它们。在上面的例子中,
weights -> kernel
谢天谢地,有人把a script which should automatically rename your variables放在一起了。这不是第一次从检查点重新命名和破坏变量。你知道吗
相关问题 更多 >
编程相关推荐