funasr/optimizers/__init__.py
@@ -0,0 +1,17 @@ import torch from funasr.optimizers.fairseq_adam import FairseqAdam from funasr.optimizers.sgd import SGD 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, )
