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),
}