prepare_qat¶ class torch.ao.quantization.prepare_qat(model, mapping=None, inplace=False)[source][source]¶ 为量化校准或量化感知训练准备模型副本,并将其转换为量化版本。 量化配置应预先分配给 .qconfig 属性中的各个子模块。 参数 model – 要就地修改的输入模型 mapping – 将浮点模块映射到要替换的量化模块的字典。 inplace – 就地执行模型转换,原始模块将被修改