diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/cycle_gan_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index 50fb505..29389db 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -197,7 +197,7 @@ class CycleGANModel(BaseModel): real_B = util.tensor2im(self.real_B.data) fake_A = util.tensor2im(self.fake_A.data) rec_B = util.tensor2im(self.rec_B.data) - if self.opt.identity > 0.0: + if self.opt.isTrain and self.opt.identity > 0.0: idt_A = util.tensor2im(self.idt_A.data) idt_B = util.tensor2im(self.idt_B.data) return OrderedDict([('real_A', real_A), ('fake_B', fake_B), ('rec_A', rec_A), ('idt_B', idt_B), |