Better Timm Integration
Currently models associated with timm requires conditional model loading logic and classifier logic in pretrained networks. We should try to remove/minimize this conditional logic so that timm models can be used seamlessly with torchvision models.
related to !49 (merged)
Edited by Ameya Shringi