恢复检查点张量流对象检测模型

时间:2018-11-20 17:59:00

标签: python tensorflow

大约一年前,我在带有多个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'

对我在做什么错有任何见解吗?我可以提供当时用来训练模型的管道配置。

0 个答案:

没有答案