I want to use two network.
And my network architecture is
My code is
self.optG.zero_grad()
self.optR.zero_grad()
refine = self.refine_network(self.cond)
loss1 = self.netG(self.gt, refine)
loss1 = loss1.sum()/int(b*c*h*w)
loss2 = self.criterion(refine, self.gt).sum() / int(b*c*h*w)
l_pix = loss1 + loss2
l_pix.backward()
self.optG.step()
self.optR.step()
Is it right code?
I don’t know how to use two network that I designed.