saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use parameter names to specify argument mapping (#13108)

Browse source 
use parameter names to specify argument mapping to avoid mismatches.
This commit is contained in:
Yufeng Li 2022-09-26 20:56:59 -07:00 committed by GitHub
parent e3bdba37a8
commit c746083344
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
onnxruntime/python/tools/quantization/calibrate.py
View file

@ -178,10 +178,10 @@ class MinMaxCalibrater(CalibraterBase):
"""
super(MinMaxCalibrater, self).__init__(
model,
op_types_to_calibrate,
augmented_model_path,
symmetric,
use_external_data_format,
op_types_to_calibrate=op_types_to_calibrate,
augmented_model_path=augmented_model_path,
symmetric=symmetric,
use_external_data_format=use_external_data_format,
)
self.intermediate_outputs = []
self.calibrate_tensors_range = None
@ -351,7 +351,11 @@ class HistogramCalibrater(CalibraterBase):
:param percentile: A float number between [0, 100]. Default 99.99.
"""
super(HistogramCalibrater, self).__init__(
model, op_types_to_calibrate, augmented_model_path, use_external_data_format
model,
op_types_to_calibrate=op_types_to_calibrate,
augmented_model_path=augmented_model_path,
symmetric=symmetric,
use_external_data_format=use_external_data_format,
)
self.intermediate_outputs = []
self.calibrate_tensors_range = None
@ -359,7 +363,6 @@ class HistogramCalibrater(CalibraterBase):
self.model_original_outputs = set(output.name for output in self.model.graph.output)
self.collector = None
self.method = method
self.symmetric = symmetric
self.num_bins = num_bins
self.num_quantized_bins = num_quantized_bins
self.percentile = percentile