diff options
| author | Jun-Yan Zhu <junyanz@users.noreply.github.com> | 2017-11-04 01:13:48 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-11-04 01:13:48 -0700 |
| commit | 5f858eb70a3c110238f74a592bad0e7be601c539 (patch) | |
| tree | a6435ed461bd036b3834a13eef42179eecf9e90a | |
| parent | a303e4b0c174c8eb86e53f79b267df2e344da81a (diff) | |
| parent | c114f1234420a51458b601761ffe4a6fd1b406a7 (diff) | |
Merge pull request #142 from strob/master
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) |