diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-22 08:15:48 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-22 08:15:48 -0700 |
| commit | 6c347282993d2e2db91b376d3113efa3774c3a22 (patch) | |
| tree | 75ede706fde3f61e73fbb9e7ea9ed6e97aff2a56 /models/pix2pix_model.py | |
| parent | a7917caaeaefe51db959b8f3ae50a20e726fbd93 (diff) | |
add dropout option for G
Diffstat (limited to 'models/pix2pix_model.py')
| -rw-r--r-- | models/pix2pix_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
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, |