autogluon.extra.model_zoo

Here we provide pretrained Models discovered via Neural Architecture Search

How To Use Pretrained Models

Example showing how to load pretrained network ‘efficientnet_b0’, which was produced via NAS.

from autogluon.extra import model_zoo
model = model_zoo.get_model('efficientnet_b0', pretrained=True)

EfficientNet

The following pretrained EfficientNet 1 models are provided for image classification. The accuracy achieved by each model on a popular image classification benchmark is indicated, along with the image crop-size used by each model.

1

Tan, Mingxing, and Quoc V. Le. “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks.

Model	Acc %	crop_size
EfficientNet_B0	77.03	224
EfficientNet_B1	78.66	240
EfficientNet_B2	79.57	260
EfficientNet_B3	80.68	300
EfficientNet_B4	81.97	380
EfficientNet_B5	83.30	456
EfficientNet_B6	83.79	528
EfficientNet_B7	83.86	600