fuse_fx

class torch.ao.quantization.quantize_fx.fuse_fx(model, fuse_custom_config=None, backend_config=None) [source]

Объединяет модули, например conv+bn, conv+bn+relu и т. д.; модель должна находиться в режиме eval. Правила объединения определены в torch.ao.quantization.fx.fusion_pattern.py

Параметры:

• model (*) – модель torch.nn.Module
• fuse_custom_config (*) – пользовательские конфигурации для fuse_fx. Подробнее см. в FuseCustomConfig

Возвращаемый тип:

GraphModule

Пример:

from torch.ao.quantization import fuse_fx

m = Model().eval()
m = fuse_fx(m)