composer.models.efficientnetb0.efficientnetb0_hparams#

YAHP interface for EfficientNetB0.

Hparams

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

EfficientNetB0Hparams

YAHP interface for EfficientNetB0.
class composer.models.efficientnetb0.efficientnetb0_hparams.EfficientNetB0Hparams(initializers=<factory>, num_classes=None, drop_connect_rate=0.2)[source]#

Bases: composer.models.model_hparams.ModelHparams

YAHP interface for EfficientNetB0.

Parameters

  • num_classes (int, optional) โ€“ The number of classes. Needed for classification tasks. Default: 1000.

  • drop_connect_rate (float, optional) โ€“ Probability of dropping a sample within a block before identity connection. Default: 0.2.