ResNetCIFARHparams#
- class composer.models.ResNetCIFARHparams(initializers=<factory>, num_classes=10, model_name=None)[source]#
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
.