游雁
2024-01-12 40d1f80030d38b3377a95ead8837e82c67aa59f6
funasr/schedulers/__init__.py
@@ -7,7 +7,7 @@
from funasr.schedulers.tri_stage_scheduler import TriStageLR
from funasr.schedulers.warmup_lr import WarmupLR
scheduler_choices = dict(
scheduler_classes = dict(
   ReduceLROnPlateau=torch.optim.lr_scheduler.ReduceLROnPlateau,
   lambdalr=torch.optim.lr_scheduler.LambdaLR,
   steplr=torch.optim.lr_scheduler.StepLR,