prepare_qat

class torch.ao.quantization.prepare_qat(model, mapping=None, inplace=False) [исходный код]

Подготавливает копию модели для калибровки квантования или обучения с учётом квантования и преобразует её в квантованную версию.

Конфигурацию квантования следует заранее назначить отдельным подмодулям в атрибуте .qconfig.

Параметры:

• model – входная модель, изменяемая на месте
• mapping – словарь, сопоставляющий модули с плавающей точкой квантованным модулям, которые следует заменить.
• inplace – выполнять преобразования модели на месте; исходный модуль будет изменён