EfficientNet
Learn to train an EfficientNet image classification model.
We'll cover the following...
We'll cover the following...
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. ...