From 4e5c4fc6cabc5155f5fd410d1a10ff447d0b186d Mon Sep 17 00:00:00 2001 From: Dark_Nemesis Date: Fri, 27 Dec 2024 18:47:47 +0530 Subject: [PATCH] Fix AttributeError in get_exec_graph_int8layers function - Convert layer_type to string before calling lower() method --- model_analyzer/model_metadata.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 = [