ResNet18

class composer.models.ResNet18(num_classes: int, initializers: Optional[List[composer.models.model_hparams.Initializer]] = None)[source]

Bases: composer.models.base.MosaicClassifier

A ResNet-18 model extending MosaicClassifier.

See this paper for details on the residual network architecture.

Parameters

num_classes (int) – The number of classes for the model.
initializers (List[Initializer], optional) – Initializers for the model. None for no initialization. (default: None)