Thanks, I will definitely give that a shot!
I have another question regarding fine tuning, I am following this tutorial: https://pytorch.org/tutorials/beginner/finetuning_torchvision_models_tutorial.html 1
I have a param_to_update tensor which I want to feed into my optim function. Currently my optimizer func is called as self.optimizer_G = torch.optim.Adam(self.netG.parameters(),
lr=opt.lr, betas=(opt.beta1, 0.999))
If I want to call the function, how do I set the netG.parameters() = param_to_update?
Sorry if my explanation is a bit confusing.