resnet_cifar_hparams#
YAHP interface for
composer_resnet_cifar()
.
Hparams
These classes are used with yahp
for YAML
-based configuration.
|
- class composer.models.resnet_cifar.resnet_cifar_hparams.ResNetCIFARHparams(initializers=<factory>, num_classes=10, model_name=None)[source]#
Bases:
composer.models.model_hparams.ModelHparams
Hparams
interface forcomposer_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
.