From 3a4281f4959534b1bf5d01acf0085f4f8e6f2ec8 Mon Sep 17 00:00:00 2001
From: wuhongsheng <664116298@qq.com>
Date: 星期五, 05 七月 2024 00:55:32 +0800
Subject: [PATCH] 优化speakid和语句匹配逻辑，部分解决speakid不从0递增问题 (#1870)

---
 funasr/optimizers/__init__.py |   26 +++++++++++++-------------
 1 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/funasr/optimizers/__init__.py b/funasr/optimizers/__init__.py
index b4dfe5d..66e2e9d 100644
--- a/funasr/optimizers/__init__.py
+++ b/funasr/optimizers/__init__.py
@@ -2,16 +2,16 @@
 from funasr.optimizers.fairseq_adam import FairseqAdam
 from funasr.optimizers.sgd import SGD
 
-optim_choices = dict(
-	adam=torch.optim.Adam,
-	fairseq_adam=FairseqAdam,
-	adamw=torch.optim.AdamW,
-	sgd=SGD,
-	adadelta=torch.optim.Adadelta,
-	adagrad=torch.optim.Adagrad,
-	adamax=torch.optim.Adamax,
-	asgd=torch.optim.ASGD,
-	lbfgs=torch.optim.LBFGS,
-	rmsprop=torch.optim.RMSprop,
-	rprop=torch.optim.Rprop,
-)
\ No newline at end of file
+optim_classes = dict(
+    adam=torch.optim.Adam,
+    fairseq_adam=FairseqAdam,
+    adamw=torch.optim.AdamW,
+    sgd=SGD,
+    adadelta=torch.optim.Adadelta,
+    adagrad=torch.optim.Adagrad,
+    adamax=torch.optim.Adamax,
+    asgd=torch.optim.ASGD,
+    lbfgs=torch.optim.LBFGS,
+    rmsprop=torch.optim.RMSprop,
+    rprop=torch.optim.Rprop,
+)

--
Gitblit v1.9.1