EfficientNetV2¶
The EfficientNetV2 model is based on the EfficientNetV2: Smaller Models and Faster Training paper.
Model builders¶
The following model builders can be used to instantiate an EfficientNetV2 model, with or
without pre-trained weights. All the model builders internally rely on the
torchvision.models.efficientnet.EfficientNet
base class. Please refer to the source
code for
more details about this class.
|
Constructs an EfficientNetV2-S architecture from EfficientNetV2: Smaller Models and Faster Training. |
|
Constructs an EfficientNetV2-M architecture from EfficientNetV2: Smaller Models and Faster Training. |
|
Constructs an EfficientNetV2-L architecture from EfficientNetV2: Smaller Models and Faster Training. |