Yabin Li
2023-12-18 65676c0aaefacff0dfbfb7e1785dda516460d9eb
funasr/optimizers/__init__.py
@@ -1,17 +0,0 @@
import torch
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,
)