diff --git a/timm/scheduler/plateau_lr.py b/timm/scheduler/plateau_lr.py index fbfc531f..cacfab3c 100644 --- a/timm/scheduler/plateau_lr.py +++ b/timm/scheduler/plateau_lr.py @@ -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)