composer.models.resnet.resnet_hparams#

composer.models.resnet.resnet_hparams

Functions

dataclass

Returns the same class as was passed in, with dunder methods added based on the fields defined in the class.

Classes

ComposerResNet

ResNet model family extending ComposerClassifier.

Hparams

These classes are used with yahp for YAML-based configuration.

`ModelHparams`	Model Hparams.
`ResNetHparams`	composer.models.resnet.resnet_hparams.ResNetHparams

class composer.models.resnet.resnet_hparams.ResNetHparams(initializers: List[composer.models.model_hparams.Initializer] = <factory>, num_classes: int = None, model_name: str = '', pretrained: bool = False, groups: int = 1, width_per_group: int = 64)[source]#

Bases: composer.models.model_hparams.ModelHparams

composer.models.resnet.resnet_hparams.ResNetHparams