From f43da18b5e097515c12438c70fc7918c29e5d0c8 Mon Sep 17 00:00:00 2001 From: lingji-yidong <75744976+lingji-yidong@users.noreply.github.com> Date: 星期一, 19 八月 2024 13:36:59 +0800 Subject: [PATCH] fix start timestamp in sentence_info (#2024) --- 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..66e2e9d 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_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