code update

shixian.shi

2024-01-15 1233c0d3ff9cf7fd6131862e7d0b208d3981f6da

 funasr/schedulers/__init__.py

@@ -8,16 +8,16 @@
from funasr.schedulers.warmup_lr import WarmupLR

scheduler_classes = dict(
   ReduceLROnPlateau=torch.optim.lr_scheduler.ReduceLROnPlateau,
   lambdalr=torch.optim.lr_scheduler.LambdaLR,
   steplr=torch.optim.lr_scheduler.StepLR,
   multisteplr=torch.optim.lr_scheduler.MultiStepLR,
   exponentiallr=torch.optim.lr_scheduler.ExponentialLR,
   CosineAnnealingLR=torch.optim.lr_scheduler.CosineAnnealingLR,
   noamlr=NoamLR,
   warmuplr=WarmupLR,
   tri_stage=TriStageLR,
   cycliclr=torch.optim.lr_scheduler.CyclicLR,
   onecyclelr=torch.optim.lr_scheduler.OneCycleLR,
   CosineAnnealingWarmRestarts=torch.optim.lr_scheduler.CosineAnnealingWarmRestarts,
    ReduceLROnPlateau=torch.optim.lr_scheduler.ReduceLROnPlateau,
    lambdalr=torch.optim.lr_scheduler.LambdaLR,
    steplr=torch.optim.lr_scheduler.StepLR,
    multisteplr=torch.optim.lr_scheduler.MultiStepLR,
    exponentiallr=torch.optim.lr_scheduler.ExponentialLR,
    CosineAnnealingLR=torch.optim.lr_scheduler.CosineAnnealingLR,
    noamlr=NoamLR,
    warmuplr=WarmupLR,
    tri_stage=TriStageLR,
    cycliclr=torch.optim.lr_scheduler.CyclicLR,
    onecyclelr=torch.optim.lr_scheduler.OneCycleLR,
    CosineAnnealingWarmRestarts=torch.optim.lr_scheduler.CosineAnnealingWarmRestarts,
)