torch.ao.quantization.quantize#

prepare

Prepares a copy of the model for quantization calibration or quantization-aware training.

convert

Converts submodules in input module to a different module according to mapping by calling from_float method on the target module class.

quantize

Quantize the input float model with post training static quantization.

prepare_qat

Prepares a copy of the model for quantization calibration or quantization-aware training and converts it to quantized version.

quantize_qat

Do quantization aware training and output a quantized model

quantize_dynamic

将浮点模型转换为动态模型(即