How can l fine tune a pre-trained model?

Hello,
Is there any tutorial that explains how to fine tune a per-trained model with new dataset

Please refer to this answer: How to perform finetuning in Pytorch?

I know, I was facing similar problems too. So after i was done, I wrote this tutorial on fine tuning a pretrained model.

https://github.com/Spandan-Madan/Pytorch_fine_tuning_Tutorial

Hope this helps!

3 Likes

I’ve updated the tutorial to work with PyTorch 0.4 now! Here is the link again - https://github.com/Spandan-Madan/Pytorch_fine_tuning_Tutorial

1 Like