diff options
| author | Jun-Yan Zhu <junyanz@users.noreply.github.com> | 2017-10-07 22:39:25 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-10-07 22:39:25 -0400 |
| commit | fe8ea339e8f8a9dac5d202ebe29f0a578f876b86 (patch) | |
| tree | e940b193953580ba75b924f85e07821e88546146 | |
| parent | 1a764f00f8fced8696592bee776d2154b1e8e5a1 (diff) | |
Update cycle_gan_model.py
| -rw-r--r-- | models/cycle_gan_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index c6b336c..6a52b2c 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, not opt.no_dropout, self.gpu_ids) + opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, opt.init_type, self.gpu_ids) self.netG_B = networks.define_G(opt.output_nc, opt.input_nc, - opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, self.gpu_ids) + opt.ngf, opt.which_model_netG, opt.norm, not opt.no_dropout, opt.init_type, self.gpu_ids) if self.isTrain: use_sigmoid = opt.no_lsgan |