diff --git a/model_analyzer/model_metadata.py b/model_analyzer/model_metadata.py index 750c108..0db1a4b 100644 --- a/model_analyzer/model_metadata.py +++ b/model_analyzer/model_metadata.py @@ -202,7 +202,7 @@ def get_exec_graph_int8layers(self, device: str = 'CPU') -> Tuple[list, list]: rt_info = execution_node.get_rt_info() layer_type = rt_info['layerType'] inputs_number = ( - 1 if layer_type.lower() in {'convolution', 'deconvolution', 'fullyconnected', 'gemm', 'pooling'} + 1 if str(layer_type).lower() in {'convolution', 'deconvolution', 'fullyconnected', 'gemm', 'pooling'} else len(execution_node.inputs()) ) input_precisions = [