diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-07-05 19:08:39 -0400 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-07-05 19:08:39 -0400 |
| commit | c7f7d1979a35b443dba7e776203ed7084efecf77 (patch) | |
| tree | 65652aaad410182eb6cb8bbc1accb8a87d75dc92 /models/cycle_gan_model.py | |
| parent | e77d1352c0618adf8abf348b04647dd86e8890c1 (diff) | |
add reflection padding layer
Diffstat (limited to 'models/cycle_gan_model.py')
| -rw-r--r-- | models/cycle_gan_model.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index 6fbb19f..8748a62 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -62,12 +62,13 @@ class CycleGANModel(BaseModel): self.optimizer_D_A = torch.optim.Adam(self.netD_A.parameters(), lr=opt.lr, betas=(opt.beta1, 0.999)) self.optimizer_D_B = torch.optim.Adam(self.netD_B.parameters(), lr=opt.lr, betas=(opt.beta1, 0.999)) - print('---------- Networks initialized -------------') - networks.print_network(self.netG_A) - networks.print_network(self.netG_B) + print('---------- Networks initialized -------------') + networks.print_network(self.netG_A) + networks.print_network(self.netG_B) + if self.isTrain: networks.print_network(self.netD_A) networks.print_network(self.netD_B) - print('-----------------------------------------------') + print('-----------------------------------------------') def set_input(self, input): AtoB = self.opt.which_direction == 'AtoB' |