From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001 From: 天地 <tiandiweizun@gmail.com> Date: 星期三, 19 三月 2025 23:10:13 +0800 Subject: [PATCH] 1. bug fix:list(mean)和list(var),由于mean和var是numpy,导致写入到文件的格式错误,参考上面的话,大概率是list(mean.tolist()),其实外层list没有必要 (#2437) --- funasr/optimizers/__init__.py | 26 +++++++++++++------------- 1 files changed, 13 insertions(+), 13 deletions(-) diff --git a/funasr/optimizers/__init__.py b/funasr/optimizers/__init__.py index b4dfe5d..66e2e9d 100644 --- a/funasr/optimizers/__init__.py +++ b/funasr/optimizers/__init__.py @@ -2,16 +2,16 @@ 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 +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