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