max
Calibrator that returns the absolute max of all collected tensors.
Classes
Max calibrator, tracks the maximum value globally. |
- class MaxCalibrator
Bases:
_Calibrator
Max calibrator, tracks the maximum value globally.
- Parameters:
calib_desc – A MaxCalibDescriptor.
num_bits – An integer. Number of bits of quantization.
axis – A tuple. see
QuantizerAttributeConfig
.unsigned – A boolean. using unsigned quantization.
- Readonly Properties:
amaxs: A list of amax. Numpy array is saved as it is likely to be used for some plot.
- __init__(num_bits=8, axis=None, unsigned=False, track_amax=False)
Initialize.
- property amaxs
Returns the list of amax`s collected so far.
- collect(x)
Tracks the absolute max of all tensors.
- Parameters:
x – A tensor
- Raises:
RuntimeError – If amax shape changes
- compute_amax()
Return the absolute max of all tensors collected.
- reset()
Reset the collected absolute max.