Solved by ptrblck in post #2 
              
              
                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 = {
   …