From 1d60960b5038653ae841370d8bf032b1b703e17a Mon Sep 17 00:00:00 2001 From: Taesung Park Date: Mon, 17 Jul 2017 10:43:08 -0700 Subject: changed the default option to use dropout --- models/cycle_gan_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'models/cycle_gan_model.py') 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 -- cgit v1.2.3-70-g09d2