composer.models.classify_mnist.model#
A simple convolutional neural network extending ComposerClassifier
.
Classes
A simple convolutional neural network extending |
|
Toy convolutional neural network architecture in pytorch for MNIST. |
- class composer.models.classify_mnist.model.MNIST_Classifier(num_classes=10, initializers=None)[source]#
Bases:
composer.models.base.ComposerClassifier
A simple convolutional neural network extending
ComposerClassifier
. This class makesModel
compatible withTrainer
- Parameters
num_classes (int, optional) โ The number of classes. Needed for classification tasks. Default:
10
initializers (List[Initializer], optional) โ list of Initializers for the model.
None
for no initialization. Default:None
Example:
from composer.models import MNIST_Classifier model = MNIST_Classifier()