summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorjunyanz <junyanz@berkeley.edu>2017-11-05 21:53:51 -0800
committerjunyanz <junyanz@berkeley.edu>2017-11-05 21:53:51 -0800
commita24e24d67d88f75869f447690f7d994fe7d42e2d (patch)
tree008b7107eb716b1d40019d9e4bfeadb323d80e8b
parent197b38ba8885483445becb10d08f9dfa2ce55fe5 (diff)
fix typo in the cyclegan
-rw-r--r--models/cycle_gan_model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py
index 71a447d..e840e7b 100644
--- a/models/cycle_gan_model.py
+++ b/models/cycle_gan_model.py
@@ -98,7 +98,7 @@ class CycleGANModel(BaseModel):
real_B = Variable(self.input_B, volatile=True)
fake_A = self.netG_B.forward(real_B)
self.rec_B = self.netG_A.forward(fake_A).data
- self.fake_A = self.fake_A.data
+ self.fake_A = fake_A.data
# get image paths
def get_image_paths(self):