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)