composer.functional

Algorithms can be used directly through our functions-based API.

from composer import functional as CF
from torchvision import models

model = models.resnet50()

# replace some layers with blurpool
CF.apply_blurpool(model)
# replace some layers with squeeze-excite
CF.apply_se(model, latent_channels=64, min_channels=128)