torch.ao.quantization.quantize.prepare_qat#
- torch.ao.quantization.quantize.prepare_qat(model, mapping=None, inplace=False)[源代码]#
Prepares a copy of the model for quantization calibration or quantization-aware training and converts it to quantized version.
Quantization configuration should be assigned preemptively to individual submodules in .qconfig attribute.
- 参数
model – input model to be modified in-place
mapping – dictionary that maps float modules to quantized modules to be replaced.
inplace – carry out model transformations in-place, the original module is mutated