composer.models.resnet20_cifar10.model#

composer.models.resnet20_cifar10.model

Classes

CIFAR10_ResNet20

A ResNet-20 model extending ComposerClassifier.

CIFAR_ResNet

A residual neural network as originally designed for CIFAR-10.

ComposerClassifier

Implements the base logic that all classifiers can build on top of.

Initializer

An enumeration.

Attributes

  • List

  • Optional

class composer.models.resnet20_cifar10.model.CIFAR10_ResNet20(num_classes=10, initializers=None)[source]#

Bases: composer.models.base.ComposerClassifier

A ResNet-20 model extending ComposerClassifier.

See this paper for details on the residual network architecture.

Parameters

  • num_classes (int) – The number of classes for the model. Default = 10.

  • initializers (List[Initializer], optional) – Initializers for the model. None for no initialization. (default: None)