From 751a018a431595b0745e864a578bfb0db6a76bc6 Mon Sep 17 00:00:00 2001 From: speech_asr <wucong.lyb@alibaba-inc.com> Date: 星期一, 15 一月 2024 17:12:58 +0800 Subject: [PATCH] update run_server.sh --- funasr/schedulers/__init__.py | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/funasr/schedulers/__init__.py b/funasr/schedulers/__init__.py index 7bb8118..cba286a 100644 --- a/funasr/schedulers/__init__.py +++ b/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, -) \ No newline at end of file +) -- Gitblit v1.9.1