File tree Expand file tree Collapse file tree
modelopt/onnx/quantization/autotune Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -291,7 +291,7 @@ def region_pattern_autotuning_workflow(
291291
292292 for region_idx , region in enumerate (regions ):
293293 logger .info (
294- f"Region { region_idx + 1 } /{ len (regions )} (ID={ region .id } , level={ region .get_level () } )"
294+ f"Region { region_idx + 1 } /{ len (regions )} (ID={ region .id } , level={ region .level } )"
295295 )
296296
297297 if node_filter_list and not _region_matches_filter (
@@ -340,7 +340,7 @@ def region_pattern_autotuning_workflow(
340340 else :
341341 logger .info (f" Tested { schemes_tested } schemes" )
342342
343- region_model_path = models_dir / f"region_{ region .id } _level_{ region .get_level () } .onnx"
343+ region_model_path = models_dir / f"region_{ region .id } _level_{ region .level } .onnx"
344344 autotuner .export_onnx (str (region_model_path ), insert_qdq = True , best = True )
345345 logger .debug (f" Saved best model: { region_model_path .name } " )
346346
You can’t perform that action at this time.
0 commit comments