fuse_fx¶
- class torch.ao.quantization.quantize_fx.fuse_fx(model, fuse_custom_config=None, backend_config=None)[源代码]¶
融合模块,例如 conv+bn、conv+bn+relu 等,模型必须处于评估模式。融合规则定义在 torch.ao.quantization.fx.fusion_pattern.py 中
- 参数
model (*) – torch.nn.Module 模型
fuse_custom_config (*) – fuse_fx 的自定义配置。有关详细信息,请参见
FuseCustomConfig
- 返回类型
示例
from torch.ao.quantization import fuse_fx m = Model().eval() m = fuse_fx(m)