Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
pre-trained model from timm model = timm.create_model('resnet50', pretrained=True) # Modify the model head for fine-tuning num_features = model.fc.in_features model.fc = nn.Linear(num_features, num_classes) Or you can modify RestNet head for classification task along with adding extra linear layers