fuse_fx¶
- class torch.ao.quantization.quantize_fx.fuse_fx(model, fuse_custom_config=None, backend_config=None)[source][source]¶
融合诸如 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)