Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
timm pretrained_model_name = "resnet50" model = timm.create_model(pretrained_model_name, pretrained=False) It is important to note that regardless of the source of the pre-trained model, the key modification required is to adjust the fully connected FC layer(or could be linear/classifier/head) of