diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index 15bb72d8ddc..03ee20ea13f 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -156,11 +156,11 @@ def backward_G(self): # Identity loss if lambda_idt > 0: # G_A should be identity if real_B is fed: ||G_A(B) - B|| - self.idt_A = self.netG_A(self.real_B) - self.loss_idt_A = self.criterionIdt(self.idt_A, self.real_B) * lambda_B * lambda_idt + self.idt_B = self.netG_A(self.real_B) + self.loss_idt_B = self.criterionIdt(self.idt_B, self.real_B) * lambda_B * lambda_idt # G_B should be identity if real_A is fed: ||G_B(A) - A|| - self.idt_B = self.netG_B(self.real_A) - self.loss_idt_B = self.criterionIdt(self.idt_B, self.real_A) * lambda_A * lambda_idt + self.idt_A = self.netG_B(self.real_A) + self.loss_idt_A = self.criterionIdt(self.idt_A, self.real_A) * lambda_A * lambda_idt else: self.loss_idt_A = 0 self.loss_idt_B = 0