update calibration.py

Signed-off-by: yuwenzho <[email protected]>

Author: yuwenzho

neural_compressor/adaptor/ox_utils/calibration.py

@@ -548,7 +548,7 @@ def dump_minmax(self, q_config):
         # pipeline of getting calib ranges of tensors during calibration:
         # 1. augment_graph(): insert activation tensors to model output
         # 2. get_intermediate_outputs():
-        #   2.1 get_activation_tensors_calib_range(): get calib ranges of activation tensors using the augmnet graph
+        #   2.1 get_activation_tensors_calib_range(): get calib ranges of activation tensors using the augment graph
         #   2.2 get_weight_tensors_calib_range(): get calib ranges of weight tensors
         self.augment_graph()
         node_output_names, output_dicts = self.get_intermediate_outputs(q_config)
@@ -631,7 +631,7 @@ def dump_tensor(self, activation=True, weight=False, format=None):
             self.dynamically_quantized = "DynamicQuantizeLinear" in [node.op_type for node in self.model.graph.node]
             is_qdq = format == "qdq"
         if activation:
-            self.augment_graph(inspect_tensor=True) # add activation tensors to model output
+            self.augment_graph() # add activation tensors to model output
         _, output_dicts = self.get_intermediate_outputs(activation_only=not weight, weight_only=not activation)
         iters = len(list(output_dicts.values())[-1])
         map_node_activation = [{} for _ in range(iters)]