add

游雁

2024-04-12 da340e6a6cf8680878a083f5c1b18775dc0c686f

 funasr/schedulers/noam_lr.py

@@ -4,7 +4,6 @@

import torch
from torch.optim.lr_scheduler import _LRScheduler
from typeguard import check_argument_types

from funasr.schedulers.abs_scheduler import AbsBatchStepScheduler

@@ -31,7 +30,6 @@
        warmup_steps: Union[int, float] = 25000,
        last_epoch: int = -1,
    ):
        assert check_argument_types()
        self.model_size = model_size
        self.warmup_steps = warmup_steps