composer.models.resnet20_cifar10.model

A ResNet-20 model extending ComposerClassifier.

Classes

CIFAR10_ResNet20

A ResNet-20 model extending ComposerClassifier.

class composer.models.resnet20_cifar10.model.CIFAR10_ResNet20(num_classes=10, initializers=None)

Bases: composer.models.base.ComposerClassifier

A ResNet-20 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_ResNet20

model = CIFAR10_ResNet20()  # creates a resnet20 for cifar image classification