Merge branch 'main' of github.com:alibaba-damo-academy/FunASR merge

游雁

2024-09-25 fc00476a25d11ee1e9c70e1a1e04c33d996b59dc

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