"在检查站中找不到张量"在使用TF-slim评估重新调整的初始-v3模型时

时间:2018-03-30 10:18:46

标签: python tensorflow

当我使用TF-slim中的eval_image_classifier.py使用我的数据集评估重新调整的inception-v3模型时,我收到了一个错误:

NotFoundError (see above for traceback): Key InceptionV3/AuxLogits/Conv2d_2a_3x3/weights not found in checkpoint
 [[Node: save/RestoreV2_7 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2_7/tensor_names, save/RestoreV2_7/shape_and_slices)]]

然后我打印了所有张量'在orignal inception-v3检查点和重新调整的检查点中的名称,然后将这些张量与张量inception_v3.arg_scope进行比较,我发现检查点中的一些张量与inception_v3.arg_scope中的张量不同:

"InceptionV3/AuxLogits/Conv2d_2a_3x3" in the inception_v3.arg_scope
"InceptionV3/AuxLogits/Conv2d_2a_5x5" in the checkpoints

之前是否改变了inception_v3的架构?如何解决这个问题呢?使用inception_v3.arg_scope重新调整整个inception-v3模型而不是重新调整?

1 个答案:

答案 0 :(得分:0)

tf.reset_default_graph()添加到python脚本,它将解决此问题。