From c114f1234420a51458b601761ffe4a6fd1b406a7 Mon Sep 17 00:00:00 2001 From: Robert M Ochshorn Date: Fri, 3 Nov 2017 09:03:19 +0100 Subject: util.tensor2im takes data, not Variable --- models/cycle_gan_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'models/cycle_gan_model.py') diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index ecb92dc..ff7330b 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -217,11 +217,11 @@ class CycleGANModel(BaseModel): def get_current_visuals(self): real_A = util.tensor2im(self.real_A.data) - fake_B = util.tensor2im(self.fake_B) - rec_A = util.tensor2im(self.rec_A) + fake_B = util.tensor2im(self.fake_B.data) + rec_A = util.tensor2im(self.rec_A.data) real_B = util.tensor2im(self.real_B.data) - fake_A = util.tensor2im(self.fake_A) - rec_B = util.tensor2im(self.rec_B) + fake_A = util.tensor2im(self.fake_A.data) + rec_B = util.tensor2im(self.rec_B.data) if self.opt.isTrain and self.opt.identity > 0.0: idt_A = util.tensor2im(self.idt_A) idt_B = util.tensor2im(self.idt_B) -- cgit v1.2.3-70-g09d2