composer.models.efficientnetb0.model#

composer.models.efficientnetb0.model

Classes

ComposerClassifier

Implements the base logic that all classifiers can build on top of.

EfficientNet

EfficientNet architecture designed for ImageNet in https://arxiv.org/abs/1905.11946.

EfficientNetB0

An EfficientNet-b0 model extending ComposerClassifier.
class composer.models.efficientnetb0.model.EfficientNetB0(num_classes, drop_connect_rate=0.2)[source]#

Bases: composer.models.base.ComposerClassifier

An EfficientNet-b0 model extending ComposerClassifier.

Based off of this paper.

Parameters

  • num_classes (int) – the number of classes in the task.

  • drop_connect_rate (float) – probability of dropping a sample within a block before identity connection.