From 2d0c8274a6dd0db9d5f7a00e71401bf7b4d65553 Mon Sep 17 00:00:00 2001 From: shixian.shi <shixian.shi@alibaba-inc.com> Date: 星期三, 10 一月 2024 11:24:06 +0800 Subject: [PATCH] update funasr-onnx --- funasr/optimizers/__init__.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/funasr/optimizers/__init__.py b/funasr/optimizers/__init__.py index b4dfe5d..177f89e 100644 --- a/funasr/optimizers/__init__.py +++ b/funasr/optimizers/__init__.py @@ -2,7 +2,7 @@ from funasr.optimizers.fairseq_adam import FairseqAdam from funasr.optimizers.sgd import SGD -optim_choices = dict( +optim_classes = dict( adam=torch.optim.Adam, fairseq_adam=FairseqAdam, adamw=torch.optim.AdamW, -- Gitblit v1.9.1