From ea06646408dfb70a5964019945bea3f0567bffd4 Mon Sep 17 00:00:00 2001 From: arvigjoka <61952580+arvigjoka@users.noreply.github.com> Date: Thu, 12 Mar 2020 15:54:46 -0400 Subject: [PATCH] Update tf_saved_model_conversion_v2.py --- .../tensorflowjs/converters/tf_saved_model_conversion_v2.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py b/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py index 81ec8ba050e..4f58c59fd22 100644 --- a/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py +++ b/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py @@ -431,7 +431,8 @@ def convert_tf_saved_model(saved_model_dir, output_graph = os.path.join( output_dir, common.ARTIFACT_MODEL_JSON_FILE_NAME) - saved_model_tags = saved_model_tags.split(',') + if saved_model_tags: + saved_model_tags = saved_model_tags.split(',') model = load(saved_model_dir, saved_model_tags) _check_signature_in_model(model, signature_def)