fuse_fx
- 类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)