From c6361cc2a7e99be802d7d7e81a93e874f0faf5cd Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 15 一月 2024 14:15:53 +0800
Subject: [PATCH] funasr1.0

---
 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