Calculate GAN and L1 loss for the generator
(self)
| 51 | self.harmonized = self.attentioned |
| 52 | |
| 53 | def backward_G(self): |
| 54 | """Calculate GAN and L1 loss for the generator""" |
| 55 | self.loss_G_L1 = self.criterionL1(self.attentioned, self.real, self.mask) * self.opt.lambda_L1 |
| 56 | self.loss_G = self.loss_G_L1 |
| 57 | self.loss_G.backward() |
| 58 | |
| 59 | def optimize_parameters(self): |
| 60 | self.forward() |