composer.models.classify_mnist.model#

composer.models.classify_mnist.model

Classes

ComposerClassifier

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

Initializer

An enumeration.

MNIST_Classifier

A simple convolutional neural network extending ComposerClassifier.

Model

Toy classifier for MNIST.

Attributes

  • List

  • Optional

  • Sequence

  • Union

class composer.models.classify_mnist.model.MNIST_Classifier(num_classes, initializers=None)[source]#

Bases: composer.models.base.ComposerClassifier

A simple convolutional neural network extending ComposerClassifier.

composer.models.MNIST_Classifier is a simple example convolutional neural network which can be used to classify MNIST data.

class composer.models.classify_mnist.model.Model(initializers, outputs)[source]#

Bases: torch.nn.modules.module.Module

Toy classifier for MNIST.

Should not be used to evaluate any method.