Rate this Page

QuantStub#

classtorch.ao.quantization.QuantStub(qconfig=None)[source]#

Quantize stub module, before calibration, this is same as an observer,it will be swapped asnnq.Quantize inconvert.

Parameters

qconfig (Optional[QConfig]) – quantization configuration for the tensor,if qconfig is not provided, we will get qconfig from parent modules