From c3442d9566f5a2011c95b0d2998958a1b5348564 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期五, 12 一月 2024 18:04:42 +0800
Subject: [PATCH] update device

---
 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