swap_module

class torch.ao.quantization.swap_module(mod, mapping, custom_module_class_mapping, use_precomputed_fake_quant=False) [исходный код]

Заменяет модуль, если у него есть квантованный аналог и к нему прикреплён observer.

Параметры:

• mod – входной модуль
• mapping – словарь, сопоставляющий модуль nn с модулем nnq

Возвращает:

Соответствующий квантованный модуль для mod