Source code for composer.models.classify_mnist.mnist_hparams
# Copyright 2021 MosaicML. All Rights Reserved.
"""`YAHP <https://docs.mosaicml.com/projects/yahp/en/stable/README.html>`_ interface for :class:`.MNIST_Classifier`."""
from dataclasses import asdict, dataclass
from composer.models.model_hparams import ModelHparams
__all__ = ["MnistClassifierHparams"]
[docs]@dataclass
class MnistClassifierHparams(ModelHparams):
"""`YAHP <https://docs.mosaicml.com/projects/yahp/en/stable/README.html>`_ interface for :class:`.MNIST_Classifier`.
Args:
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``.
"""
def initialize_object(self):
from composer.models import MNIST_Classifier
return MNIST_Classifier(**asdict(self))