diff options
| -rw-r--r-- | models/cycle_gan_model.py | 1 | ||||
| -rw-r--r-- | models/pix2pix_model.py | 1 |
2 files changed, 0 insertions, 2 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index bcc6a15..730d077 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -40,7 +40,6 @@ class CycleGANModel(BaseModel): self.load_network(self.netD_B, 'D_B', which_epoch) if self.isTrain: - self.old_lr = opt.lr self.fake_A_pool = ImagePool(opt.pool_size) self.fake_B_pool = ImagePool(opt.pool_size) # define loss functions diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 9c46a19..78f8d69 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -30,7 +30,6 @@ class Pix2PixModel(BaseModel): if self.isTrain: self.fake_AB_pool = ImagePool(opt.pool_size) - self.old_lr = opt.lr # define loss functions self.criterionGAN = networks.GANLoss(use_lsgan=not opt.no_lsgan, tensor=self.Tensor) self.criterionL1 = torch.nn.L1Loss() |