From a6607a17ebead422a55091552c94f45b14225377 Mon Sep 17 00:00:00 2001 From: shikha-gist <43250652+shikha-gist@users.noreply.github.com> Date: Mon, 6 May 2024 16:27:01 -0600 Subject: [PATCH] Update cycle_gan_model.py --- models/cycle_gan_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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