EfficientNet

Learn to train an EfficientNet image classification model.

Supported variants

The PyTorch Image Model supports the following EfficientNet variants:

Base model Resolution
EfficientNetB0 224
EfficientNetB1 240
EfficientNetB2 260
EfficientNetB3 300
EfficientNetB4 380

EfficientNetB0 is a good start for most use cases. Other variants require higher GPU memory since the base image resolution is higher. ...