Input size for EfficientNet versions from torchvision.models

Yes, the torchvision EfficienNet models also expect a different input shape as described in the docs:

The sizes of the EfficientNet models depend on the variant. For the exact input sizes check here

where the link points to:

    elif args.model.startswith('efficientnet_'):
        sizes = {
            'b0': (256, 224), 'b1': (256, 240), 'b2': (288, 288), 'b3': (320, 300),
            'b4': (384, 380), 'b5': (489, 456), 'b6': (561, 528), 'b7': (633, 600),
        }
2 Likes