KeyError in freeze model

时间:2018-11-13 07:50:54

标签: python tensorflow

I've used the TensorFlow's graph_util.convert_variables_to_constants function to freeze a ckpt model. I have added all nodes in the ckpt model to the whitelist of the convert_variables_to_constants.

However, when I want to load the freezed model and use it, apparently a few nodes have not been added. I mean in the bellow line:

logits = tf.get_default_graph().get_tensor_by_name("Logits/BiasAdd:0")

it gives me the following error:

KeyError: "The name 'Logits/BiasAdd:0' refers to a Tensor which does not exist. The operation, 'Logits/BiasAdd', does not exist in the graph."

Any ideas how can I add this node to the freezed model?

0 个答案:

没有答案