ResNetCIFARHparams#

class composer.models.ResNetCIFARHparams(initializers=<factory>, num_classes=10, model_name=None)[source]#

Hparams interface for composer_resnet_cifar().

Parameters

  • model_name (str) โ€“ "resnet_9", "resnet_20", or "resnet_56".

  • 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.