From c8c45036724e232a63f906d7881c1b8622051d13 Mon Sep 17 00:00:00 2001 From: Anastasiia Pnevskaia Date: Mon, 20 Feb 2023 15:20:54 +0100 Subject: [PATCH] Error messages correcting in MO extractor (#15783) * Error messages correcting. * Error messages correcting. * Small corrections. --- tools/mo/openvino/tools/mo/moc_frontend/extractor.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tools/mo/openvino/tools/mo/moc_frontend/extractor.py b/tools/mo/openvino/tools/mo/moc_frontend/extractor.py index b3d69d72ff4..093067b60c5 100644 --- a/tools/mo/openvino/tools/mo/moc_frontend/extractor.py +++ b/tools/mo/openvino/tools/mo/moc_frontend/extractor.py @@ -236,11 +236,13 @@ def fe_input_user_data_repack( # NOTE: the logic relies on parameters with the single name model_inputs = input_model.get_inputs() frozen_names = freeze_placeholder.keys() - assert len(model_inputs) == len(frozen_names) + 1, "Please check the conversion command-line. " \ - "Total number of model inputs must match to a number " \ - "of input shapes along with frozen inputs." + assert len(model_inputs) == len(frozen_names) + 1, \ + "Please check the conversion command-line. Total number of model inputs ({} detected) " \ + "must match to a number of input shapes along with frozen inputs ({} in total).".format( + len(model_inputs), + len(frozen_names) + 1) for node in model_inputs: - assert len(node.get_names()) > 0, "Original input models must have names." + assert len(node.get_names()) > 0, "Original model inputs must have tensor names." input_name = node.get_names()[0] if input_name not in frozen_names: _input_shapes.append( @@ -258,7 +260,7 @@ def fe_input_user_data_repack( # NOTE: the logic relies on parameters with the single name assert input_user_shapes is None for node in input_model.get_inputs(): - assert len(node.get_names()) > 0, "Original input models must have names." + assert len(node.get_names()) > 0, "Original model inputs must have tensor names." input_name = node.get_names()[0] _input_shapes.append( {