From ddbc8b5eded1fff6084001d160d46b532020ecb7 Mon Sep 17 00:00:00 2001 From: Shi Xian <40013335+R1ckShi@users.noreply.github.com> Date: 星期一, 15 一月 2024 20:36:20 +0800 Subject: [PATCH] Merge pull request #1247 from alibaba-damo-academy/funasr1.0 --- funasr/optimizers/__init__.py | 22 +++++++++++----------- 1 files changed, 11 insertions(+), 11 deletions(-) diff --git a/funasr/optimizers/__init__.py b/funasr/optimizers/__init__.py index 177f89e..a1a57a5 100644 --- a/funasr/optimizers/__init__.py +++ b/funasr/optimizers/__init__.py @@ -3,15 +3,15 @@ from funasr.optimizers.sgd import SGD 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, + 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