From 6c347282993d2e2db91b376d3113efa3774c3a22 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 22 Apr 2017 08:15:48 -0700 Subject: add dropout option for G --- models/pix2pix_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/pix2pix_model.py') diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 3bdd237..0e02ebf 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -25,7 +25,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, self.gpu_ids) + opt.which_model_netG, opt.norm, opt.use_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, -- cgit v1.2.3-70-g09d2