From 349614a2f168654ba59bf1461ea61e1cb9358eb6 Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 2 May 2017 20:26:39 -0700 Subject: use InstanceNorm2d from the latest pytorch --- 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 451002d..f8c4f9f 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -35,10 +35,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, use_sigmoid, self.gpu_ids) + opt.n_layers_D, opt.norm, use_sigmoid, self.gpu_ids) self.netD_B = networks.define_D(opt.input_nc, opt.ndf, opt.which_model_netD, - opt.n_layers_D, use_sigmoid, self.gpu_ids) + opt.n_layers_D, opt.norm, use_sigmoid, 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