diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-23 04:29:54 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-23 04:29:54 -0700 |
| commit | d5a38496cf624cb110cb292d18e0822b826b61ab (patch) | |
| tree | 12e88625c75ead378e6cbf2b0dee8c2cf09166e2 /models/cycle_gan_model.py | |
| parent | 5bf446e4007f9b1c657c32f817e78437afb35c04 (diff) | |
fix which_direction bug
Diffstat (limited to 'models/cycle_gan_model.py')
| -rw-r--r-- | models/cycle_gan_model.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index d361e47..451002d 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -2,7 +2,6 @@ import numpy as np import torch import os from collections import OrderedDict -from pdb import set_trace as st from torch.autograd import Variable import itertools import util.util as util @@ -72,7 +71,7 @@ class CycleGANModel(BaseModel): print('-----------------------------------------------') def set_input(self, input): - AtoB = self.opt.which_direction is 'AtoB' + AtoB = self.opt.which_direction == 'AtoB' input_A = input['A' if AtoB else 'B'] input_B = input['B' if AtoB else 'A'] self.input_A.resize_(input_A.size()).copy_(input_A) |