diff options
| author | Anders Christiansen <andersasac@gmail.com> | 2017-10-09 14:30:05 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-10-09 14:30:05 -0400 |
| commit | fdbe12f4eab2de8b74e5c73694ee518541189598 (patch) | |
| tree | f9a375c0014d28b143af130ceadcf203d3b99369 /models/cycle_gan_model.py | |
| parent | fe8ea339e8f8a9dac5d202ebe29f0a578f876b86 (diff) | |
Update cycle_gan_model.py
Fix argument mismatch. Was feeding gpu_ids to the init_type argument
Diffstat (limited to 'models/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 6a52b2c..50fb505 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -36,10 +36,10 @@ class CycleGANModel(BaseModel): use_sigmoid = opt.no_lsgan self.netD_A = networks.define_D(opt.output_nc, opt.ndf, opt.which_model_netD, - opt.n_layers_D, opt.norm, use_sigmoid, self.gpu_ids) + opt.n_layers_D, opt.norm, use_sigmoid, opt.init_type, self.gpu_ids) self.netD_B = networks.define_D(opt.input_nc, opt.ndf, opt.which_model_netD, - opt.n_layers_D, opt.norm, use_sigmoid, self.gpu_ids) + opt.n_layers_D, opt.norm, use_sigmoid, opt.init_type, self.gpu_ids) if not self.isTrain or opt.continue_train: which_epoch = opt.which_epoch self.load_network(self.netG_A, 'G_A', which_epoch) |