composer.models.efficientnetb0.model#
A ComposerClassifier
wrapper around the EfficientNet-b0 architecture.
Classes
A |
- class composer.models.efficientnetb0.model.EfficientNetB0(num_classes=1000, drop_connect_rate=0.2)[source]#
Bases:
composer.models.base.ComposerClassifier
A
ComposerClassifier
wrapper around the EfficientNet-b0 architecture. From Rethinking Model Scaling for Convolutional Neural Networks (Tan et al, 2019).- Parameters
Example:
from composer.models import EfficientNetB0 model = EfficientNetB0() # creates EfficientNet-b0 for image classification