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