From 2f949efbee4e2ff88dd28229d29bd6e76b75f16d Mon Sep 17 00:00:00 2001 From: Bin Li Date: Fri, 11 Oct 2019 14:49:03 +0800 Subject: [PATCH] Fix multiple models in one yml --- tools/python/transform/tensorflow_converter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/python/transform/tensorflow_converter.py b/tools/python/transform/tensorflow_converter.py index 3fe61d47..dfb41a25 100644 --- a/tools/python/transform/tensorflow_converter.py +++ b/tools/python/transform/tensorflow_converter.py @@ -310,6 +310,8 @@ class TensorflowConverter(base_converter.ConverterInterface): self.add_shape_info(transformed_graph_def) + # reset default graph to clear earlier import + tf.reset_default_graph() with tf.Session() as session: with session.graph.as_default() as graph: tf.import_graph_def(transformed_graph_def, name='') @@ -319,7 +321,6 @@ class TensorflowConverter(base_converter.ConverterInterface): # we have polluted graph with 'shape' ops, so reset it and reload it # again tf.reset_default_graph() - with tf.Session() as session: with session.graph.as_default() as graph: tf.import_graph_def(transformed_graph_def, name='') -- GitLab