From fdbe12f4eab2de8b74e5c73694ee518541189598 Mon Sep 17 00:00:00 2001 From: Anders Christiansen Date: Mon, 9 Oct 2017 14:30:05 -0400 Subject: Update cycle_gan_model.py Fix argument mismatch. Was feeding gpu_ids to the init_type argument --- 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 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) -- cgit v1.2.3-70-g09d2