Remove redundant noise attr from Plateau scheduler (use parent)

pull/1187/head
Ross Wightman 3 years ago
parent 7514439573
commit 341b464a5a

@ -30,7 +30,16 @@ class PlateauLRScheduler(Scheduler):
noise_seed=None,
initialize=True,
):
super().__init__(optimizer, 'lr', initialize=initialize)
super().__init__(
optimizer,
'lr',
noise_range_t=noise_range_t,
noise_type=noise_type,
noise_pct=noise_pct,
noise_std=noise_std,
noise_seed=noise_seed,
initialize=initialize,
)
self.lr_scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(
self.optimizer,
@ -43,11 +52,6 @@ class PlateauLRScheduler(Scheduler):
min_lr=lr_min
)
self.noise_range_t = noise_range_t
self.noise_pct = noise_pct
self.noise_type = noise_type
self.noise_std = noise_std
self.noise_seed = noise_seed if noise_seed is not None else 42
self.warmup_t = warmup_t
self.warmup_lr_init = warmup_lr_init
if self.warmup_t:
@ -85,7 +89,6 @@ class PlateauLRScheduler(Scheduler):
if self._is_apply_noise(epoch):
self._apply_noise(epoch)
def _apply_noise(self, epoch):
noise = self._calculate_noise(epoch)

Loading…
Cancel
Save