zhifu gao
2024-04-24 861147c7308b91068ffa02724fdf74ee623a909e
funasr/schedulers/tri_stage_scheduler.py
@@ -42,9 +42,7 @@
        self.decay_steps = int(self.max_update * self.phase_ratio[2])
        self.warmup_rate = (
            (self.peak_lr - self.init_lr) / self.warmup_steps
            if self.warmup_steps != 0
            else 0
            (self.peak_lr - self.init_lr) / self.warmup_steps if self.warmup_steps != 0 else 0
        )
        self.decay_factor = -math.log(self.final_lr_scale) / self.decay_steps