composer.models.resnet56_cifar10.resnet56_cifar10_hparams#

YAHP interface for CIFAR10_ResNet56.

Hparams

These classes are used with yahp for YAML-based configuration.

CIFARResNetHparams

YAHP interface for CIFAR10_ResNet56.
class composer.models.resnet56_cifar10.resnet56_cifar10_hparams.CIFARResNetHparams(initializers=<factory>, num_classes=None)[source]#

Bases: composer.models.model_hparams.ModelHparams

YAHP interface for CIFAR10_ResNet56.

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.