From 172e7ac986f299ad545cbd91a8cecc3ef967af36 Mon Sep 17 00:00:00 2001 From: zhifu gao <zhifu.gzf@alibaba-inc.com> Date: 星期一, 11 十二月 2023 10:17:22 +0800 Subject: [PATCH] Revert "Dev gzf funasr2" (#1164) --- funasr/schedulers/__init__.py | 23 ----------------------- 1 files changed, 0 insertions(+), 23 deletions(-) diff --git a/funasr/schedulers/__init__.py b/funasr/schedulers/__init__.py index 7bb8118..e69de29 100644 --- a/funasr/schedulers/__init__.py +++ b/funasr/schedulers/__init__.py @@ -1,23 +0,0 @@ -import torch -import torch.multiprocessing -import torch.nn -import torch.optim - -from funasr.schedulers.noam_lr import NoamLR -from funasr.schedulers.tri_stage_scheduler import TriStageLR -from funasr.schedulers.warmup_lr import WarmupLR - -scheduler_choices = 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, -) \ No newline at end of file -- Gitblit v1.9.1