How to perform finetuning in Pytorch?

Thanks :slight_smile: It helped.