1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
| 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,
| )
|
|
