composer.models.efficientnetb0.efficientnetb0_hparams#

composer.models.efficientnetb0.efficientnetb0_hparams

Functions

dataclass

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

Hparams

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

EfficientNetB0Hparams

composer.models.efficientnetb0.efficientnetb0_hparams.EfficientNetB0Hparams

ModelHparams

Model Hparams.

class composer.models.efficientnetb0.efficientnetb0_hparams.EfficientNetB0Hparams(initializers: List[composer.models.model_hparams.Initializer] = <factory>, num_classes: Union[int, NoneType] = None, drop_connect_rate: float = 0.2)[source]#

Bases: composer.models.model_hparams.ModelHparams

composer.models.efficientnetb0.efficientnetb0_hparams.EfficientNetB0Hparams