composer.models.resnet56_cifar10.model#

A ResNet-56 model extending ComposerClassifier.

Classes

CIFAR10_ResNet56

A ResNet-56 model extending ComposerClassifier.
class composer.models.resnet56_cifar10.model.CIFAR10_ResNet56(num_classes=10, initializers=None)[source]#

Bases: composer.models.base.ComposerClassifier

A ResNet-56 model extending ComposerClassifier.

From Deep Residual Learning for Image Recognition (He et al, 2015).

Parameters

  • num_classes (int, optional) โ€“ The number of classes. Needed for classification tasks. Default: 10.

  • initializers (List[Initializer], optional) โ€“ Initializers for the model. None for no initialization. Default: None.

Example:

from composer.models import CIFAR10_ResNet56

model = CIFAR10_ResNet56()  # creates a resnet56 for cifar image classification