Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
functions too) nn.Dropout(0.5), # Dropout layer with 50% probability nn.Linear(256, num_classes) # Final prediction fc layer ) Or you can modify RestNet head for regression task: model = timm.create_model('resnet50', pretrained=True) # Modify the model head for regression num_features =