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. ...