MnistClassifierHparams

class composer.models.MnistClassifierHparams(initializers=<factory>, num_classes=None)

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.