composer.models.timm.timm_hparams#

composer.models.timm.timm_hparams

Functions

dataclass

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

Classes

Timm

A wrapper around timm.create_model() used to create a ComposerClassifier from a timm model.

Hparams

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

`ModelHparams`	Model Hparams.
`TimmHparams`	composer.models.timm.timm_hparams.TimmHparams

Attributes

Optional

class composer.models.timm.timm_hparams.TimmHparams(initializers: List[composer.models.model_hparams.Initializer] = <factory>, num_classes: int = 1000, model_name: str = None, pretrained: bool = False, drop_rate: float = 0.0, drop_path_rate: Union[float, NoneType] = None, drop_block_rate: Union[float, NoneType] = None, global_pool: Union[str, NoneType] = None, bn_momentum: Union[float, NoneType] = None, bn_eps: Union[float, NoneType] = None)[source]#

Bases: composer.models.model_hparams.ModelHparams

composer.models.timm.timm_hparams.TimmHparams