composer.models.resnet_cifar.model#
ResNet models for CIFAR extending ComposerClassifier
.
Classes
ResNet models for CIFAR10 extending |
- class composer.models.resnet_cifar.model.ComposerResNetCIFAR(model_name, num_classes=10, initializers=None)[source]#
Bases:
composer.models.tasks.classification.ComposerClassifier
ResNet models for CIFAR10 extending
ComposerClassifier
.From Deep Residual Learning for Image Recognition (He et al, 2015). ResNet9 is based on the model from myrtle.ai blog.
- 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
.
Example:
from composer.models import ComposerResNetCIFAR model = ComposerResNetCIFAR(model_name="resnet_56") # creates a resnet56 for cifar image classification