MNIST_Classifier

class composer.models.MNIST_Classifier(num_classes: int, initializers: Optional[List[composer.models.model_hparams.Initializer]] = None)[source]

Bases: composer.models.base.MosaicClassifier

A simple convolutional neural network.

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