From 27f31cd42bb4e20dc19de0034fc0d80b449f1db1 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 06 十二月 2023 17:01:12 +0800
Subject: [PATCH] funasr2

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

diff --git a/funasr/optimizers/__init__.py b/funasr/optimizers/__init__.py
index e69de29..b4dfe5d 100644
--- a/funasr/optimizers/__init__.py
+++ b/funasr/optimizers/__init__.py
@@ -0,0 +1,17 @@
+import torch
+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

--
Gitblit v1.9.1