游雁
2024-01-14 8912e0696af069de47646fdb8a9d9c4e086e88b3
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,
@@ -20,4 +20,4 @@
   cycliclr=torch.optim.lr_scheduler.CyclicLR,
   onecyclelr=torch.optim.lr_scheduler.OneCycleLR,
   CosineAnnealingWarmRestarts=torch.optim.lr_scheduler.CosineAnnealingWarmRestarts,
)
)