From bd893b38d158d0b663321052c24dc1f4acccf552 Mon Sep 17 00:00:00 2001 From: shikha-gist <43250652+shikha-gist@users.noreply.github.com> Date: Mon, 6 May 2024 16:14:06 -0600 Subject: [PATCH] Update cycle_gan_model.py --- models/cycle_gan_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index 15bb72d8ddc..6606438a88e 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -56,7 +56,7 @@ def __init__(self, opt): # specify the images you want to save/display. The training/test scripts will call visual_names_A = ['real_A', 'fake_B', 'rec_A'] visual_names_B = ['real_B', 'fake_A', 'rec_B'] - if self.isTrain and self.opt.lambda_identity > 0.0: # if identity loss is used, we also visualize idt_B=G_A(B) ad idt_A=G_A(B) + if self.isTrain and self.opt.lambda_identity > 0.0: # if identity loss is used, we also visualize idt_B=G_A(B) ad idt_A=G_B(A) visual_names_A.append('idt_B') visual_names_B.append('idt_A')