diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-07-03 17:18:13 -0400 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-07-03 17:18:13 -0400 |
| commit | 233630e79d79901faff420eb0ae481b35d952f97 (patch) | |
| tree | 66b98747d7c0a97b37e2921ecbc378ae994aef35 /models/pix2pix_model.py | |
| parent | 11690eaffc7dcdc0f64267263f5d7a3b4fc735cf (diff) | |
fix instancenorm & batchnorm
Diffstat (limited to 'models/pix2pix_model.py')
| -rw-r--r-- | models/pix2pix_model.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index e44529b..3ab45fd 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -48,10 +48,11 @@ class Pix2PixModel(BaseModel): self.optimizer_D = torch.optim.Adam(self.netD.parameters(), lr=opt.lr, betas=(opt.beta1, 0.999)) - print('---------- Networks initialized -------------') - networks.print_network(self.netG) + print('---------- Networks initialized -------------') + networks.print_network(self.netG) + if self.isTrain: networks.print_network(self.netD) - print('-----------------------------------------------') + print('-----------------------------------------------') def set_input(self, input): AtoB = self.opt.which_direction == 'AtoB' |