Remove redundant noise attr from Plateau scheduler (use parent)
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:
|
||||
|
@ -84,7 +88,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…
Reference in New Issue