mnist_hparams#
YAHP interface for mnist_model()
.
Hparams
These classes are used with yahp
for YAML
-based configuration.
YAHP interface for |
- class composer.models.classify_mnist.mnist_hparams.MnistClassifierHparams(initializers=<factory>, num_classes=None)[source]#
Bases:
composer.models.model_hparams.ModelHparams
YAHP interface for
mnist_model()
.- 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
.