diff options
| author | Robert M Ochshorn <mail@RMOZONE.COM> | 2017-11-03 09:03:19 +0100 |
|---|---|---|
| committer | Robert M Ochshorn <mail@RMOZONE.COM> | 2017-11-03 09:03:19 +0100 |
| commit | c114f1234420a51458b601761ffe4a6fd1b406a7 (patch) | |
| tree | a6435ed461bd036b3834a13eef42179eecf9e90a | |
| parent | a303e4b0c174c8eb86e53f79b267df2e344da81a (diff) | |
util.tensor2im takes data, not Variable
| -rw-r--r-- | models/cycle_gan_model.py | 8 |
1 files changed, 4 insertions, 4 deletions
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) |