| | |
|---|
| | | from funasr.schedulers.noam_lr import NoamLR |
|---|
| | | from funasr.schedulers.tri_stage_scheduler import TriStageLR |
|---|
| | | from funasr.schedulers.warmup_lr import WarmupLR |
|---|
| | | from funasr.schedulers.lambdalr_cus import CustomLambdaLR |
|---|
| | | |
|---|
| | | scheduler_classes = dict( |
|---|
| | | ReduceLROnPlateau=torch.optim.lr_scheduler.ReduceLROnPlateau, |
|---|
| | |
|---|
| | | cycliclr=torch.optim.lr_scheduler.CyclicLR, |
|---|
| | | onecyclelr=torch.optim.lr_scheduler.OneCycleLR, |
|---|
| | | CosineAnnealingWarmRestarts=torch.optim.lr_scheduler.CosineAnnealingWarmRestarts, |
|---|
| | | custom_lambdalr=CustomLambdaLR, |
|---|
| | | ) |
|---|
