composer.models.resnet

Modules

composer.models.resnet.model	A ComposerClassifier wrapper around the torchvision implementations of the ResNet model family.
composer.models.resnet.resnet_hparams	YAHP interface for ComposerResNet.

The ResNet model family is a set of convolutional neural networks described in Deep Residual Learning for Image Recognition (He et al, 2015). ResNets can be used as the base for a variety of vision tasks. ImageNet ResNets are a subset of the ResNet family which were designed specifically for classification on the ImageNet dataset.

See the Model Card for more details.

Classes

ComposerResNet

A ComposerClassifier wrapper around the torchvision implementations of the ResNet model family.

Hparams

These classes are used with yahp for YAML-based configuration.

ResNetHparams

YAHP interface for ComposerResNet.