`neural_compressor.onnxrt.algorithms.smoother.calibrator`

Calibration for smooth quant.

Module Contents

Classes

Dump information for smooth quant.

class neural_compressor.onnxrt.algorithms.smoother.calibrator.Calibrator(model: neural_compressor.onnxrt.utils.onnx_model.ONNXModel, dataloader: neural_compressor.onnxrt.quantization.calibrate.CalibrationDataReader, iterations: List[int] = [], providers: List[str] = ['CPUExecutionProvider'], **kwargs)[source]: Dump information for smooth quant.