From 4d907718f39e2b0f7a0c714c2e3de289e742fc61 Mon Sep 17 00:00:00 2001
From: Carl <415692979@qq.com>
Date: 星期四, 28 三月 2024 13:42:00 +0800
Subject: [PATCH] 修正commit 87b62d68957a2194b017a43b6c2a15424a05a984 引入的英文整句标点预测导致末尾两个单词中间的空格被删除的问题。 (#1556)

---
 funasr/schedulers/__init__.py |   26 ++++++++++++++------------
 1 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/funasr/schedulers/__init__.py b/funasr/schedulers/__init__.py
index cba286a..39f8c0e 100644
--- a/funasr/schedulers/__init__.py
+++ b/funasr/schedulers/__init__.py
@@ -6,18 +6,20 @@
 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,
-	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,
+    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,
+    custom_lambdalr=CustomLambdaLR,
 )

--
Gitblit v1.9.1