Update core.py

Signed-off-by: Wang, Mengni <[email protected]>
onnx · Jul 23, 2024 · be276c6 · be276c6
1 parent f528f03
commit be276c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx_neural_compressor/algorithms/layer_wise/core.py b/onnx_neural_compressor/algorithms/layer_wise/core.py
@@ -121,7 +121,7 @@ def layer_wise_quant(
                 split_model_part_1.model, complete_data_reader
             )
 
-            # next_data_reader contains split_model_part_1 output data
+            # complete_data_reader contains split_model_part_1 output data
             complete_data_reader = _prepare_data_reader_for_next_split_model(
                 split_model_part_1.model_path,
                 [i.name for i in split_model_part_2.model.graph.input],