diff options
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 34e0bac..4581d33 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -28,7 +28,7 @@ class Pix2PixModel(BaseModel): use_sigmoid = opt.no_lsgan self.netD = networks.define_D(opt.input_nc + 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) if not self.isTrain or opt.continue_train: self.load_network(self.netG, 'G', opt.which_epoch) if self.isTrain: |