大约一年前,我在带有多个GPU的主机上微调了对象检测模型几天。直到现在我还没有机会使用它,但是我不知道如何加载它。
我有许多检查点文件:
model.ckpt-400000.data-00000-of-00001
model.ckpt-400000.index
model.ckpt-400000.meta
在目录checkpoints
中。我一直试图像这样加载它们:
with tf.Session() as sess:
saver = tf.train.import_meta_graph("checkpoints/model.ckpt-400000.meta")
saver.restore(sess, "checkpoints/model.ckpt-400000")
但是,这给出了例外:
.virtualenv/lib/python2.7/site-packages/tensorflow/python/framework/importer.pyc in _RemoveDefaultAttrs(op_dict, producer_op_list, graph_def)
209 # Remove any default attr values that aren't in op_def.
210 if node.op in producer_op_dict:
--> 211 op_def = op_dict[node.op]
212 producer_op_def = producer_op_dict[node.op]
213 # We make a copy of node.attr to iterate through since we may modify
KeyError: u'ClipByValue'
对我在做什么错有任何见解吗?我可以提供当时用来训练模型的管道配置。