diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-07-17 10:43:08 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-07-17 10:43:08 -0700 |
| commit | 1d60960b5038653ae841370d8bf032b1b703e17a (patch) | |
| tree | 36e7431bd02df44648cd196d15d78ef03fffda1e /models | |
| parent | 23eb65de64e62141258acc258c930c6a63daaf42 (diff) | |
changed the default option to use dropout
Diffstat (limited to 'models')
| -rw-r--r-- | models/cycle_gan_model.py | 4 | ||||
| -rw-r--r-- | models/pix2pix_model.py | 2 | ||||
| -rw-r--r-- | models/test_model.py | 2 |
3 files changed, 4 insertions, 4 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index 8748a62..b3c52c7 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -28,9 +28,9 @@ class CycleGANModel(BaseModel): # Code (paper): G_A (G), G_B (F), D_A (D_Y), D_B (D_X) self.netG_A = networks.define_G(opt.input_nc, opt.output_nc, - opt.ngf, opt.which_model_netG, opt.norm, opt.use_dropout, self.gpu_ids) + opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, self.gpu_ids) self.netG_B = networks.define_G(opt.output_nc, opt.input_nc, - opt.ngf, opt.which_model_netG, opt.norm, opt.use_dropout, self.gpu_ids) + opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, self.gpu_ids) if self.isTrain: use_sigmoid = opt.no_lsgan diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 3ab45fd..a524f2c 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -24,7 +24,7 @@ class Pix2PixModel(BaseModel): # load/define networks self.netG = networks.define_G(opt.input_nc, opt.output_nc, opt.ngf, - opt.which_model_netG, opt.norm, opt.use_dropout, self.gpu_ids) + opt.which_model_netG, opt.norm, not opt.no_dropout, self.gpu_ids) if self.isTrain: use_sigmoid = opt.no_lsgan self.netD = networks.define_D(opt.input_nc + opt.output_nc, opt.ndf, diff --git a/models/test_model.py b/models/test_model.py index 65aa088..03aef65 100644 --- a/models/test_model.py +++ b/models/test_model.py @@ -16,7 +16,7 @@ class TestModel(BaseModel): self.netG = networks.define_G(opt.input_nc, opt.output_nc, opt.ngf, opt.which_model_netG, - opt.norm, opt.use_dropout, + opt.norm, not opt.no_dropout, self.gpu_ids) which_epoch = opt.which_epoch self.load_network(self.netG, 'G', which_epoch) |