Can you please help me in making changes for the training loop. I will be really thankful to you.
Actually I followed this link for setting up the trainng loop for my model .
https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html