Shortcuts

fuse_fx

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

Fuse modules like conv+bn, conv+bn+relu etc, model must be in eval mode. Fusion rules are defined in torch.quantization.fx.fusion_pattern.py

Parameters:
  • model (*) – a torch.nn.Module model

  • fuse_custom_config (*) – custom configurations for fuse_fx. See FuseCustomConfig for more details

Return type:

GraphModule

Example:

from torch.ao.quantization import fuse_fx
m = Model().eval()
m = fuse_fx(m)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources