From c9c95ec30aba4e114003d5196c06bb8ae5ca8cc4 Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期四, 28 十二月 2023 14:36:52 +0800 Subject: [PATCH] funasr1.0 --- funasr/schedulers/__init__.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/funasr/schedulers/__init__.py b/funasr/schedulers/__init__.py index 7bb8118..2ee3a9e 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, -- Gitblit v1.9.1