Fine-Tune with a Lower Learning Rate
Gently update pretrained weights.
Beyond Just the Head
Freezing the backbone is fast, but sometimes you want the whole model to adapt. Fine-tuning unfreezes it and trains everything together. 🔧
Unfreeze the Backbone
To fine-tune, you flip requires_grad back to True so the pretrained weights can move again during training.
for p in model.parameters():
p.requires_grad = TrueAll lessons in this course
- Freeze the Backbone, Train the Head
- Fine-Tune with a Lower Learning Rate
- Discriminative Layer-Wise Rates
- Fine-Tune a Hugging Face Model