QuantStub¶
- class torch.ao.quantization.QuantStub(qconfig=None)[source][source]¶
Quantize stub module, before calibration, this is same as an observer, it will be swapped as nnq.Quantize in convert.
- Parameters
qconfig – quantization configuration for the tensor, if qconfig is not provided, we will get qconfig from parent modules
