From abc6a336b1c8ae5de5f9f0cfb0a404b9bc5c984e Mon Sep 17 00:00:00 2001 From: Vladimir Gavrilov Date: Thu, 22 Jul 2021 14:39:54 +0300 Subject: [PATCH] Fixed processing tags in summarize_graph.py (#6726) --- model-optimizer/mo/utils/summarize_graph.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/model-optimizer/mo/utils/summarize_graph.py b/model-optimizer/mo/utils/summarize_graph.py index 2a652ba3a76ad2..29180d8c616965 100644 --- a/model-optimizer/mo/utils/summarize_graph.py +++ b/model-optimizer/mo/utils/summarize_graph.py @@ -70,9 +70,10 @@ def summarize_graph(graph_def): if argv.input_model and argv.saved_model_dir: print("[ ERROR ] Both keys were provided --input_model and --input_dir. Please, provide only one of them") sys.exit(1) - graph_def, _ = load_tf_graph_def(graph_file_name=argv.input_model, is_binary=not argv.text, - checkpoint=argv.input_checkpoint, - model_dir=argv.saved_model_dir, saved_model_tags=argv.saved_model_tags) + tags = argv.saved_model_tags.split(",") + graph_def, _, _ = load_tf_graph_def(graph_file_name=argv.input_model, is_binary=not argv.text, + checkpoint=argv.input_checkpoint, + model_dir=argv.saved_model_dir, saved_model_tags=tags) summary = summarize_graph(graph_def) print("{} input(s) detected:".format(len(summary['inputs']))) for input in summary['inputs']: