shixian.shi
2024-01-12 c3442d9566f5a2011c95b0d2998958a1b5348564
funasr/optimizers/__init__.py
@@ -2,7 +2,7 @@
from funasr.optimizers.fairseq_adam import FairseqAdam
from funasr.optimizers.sgd import SGD
optim_choices = dict(
optim_classes = dict(
   adam=torch.optim.Adam,
   fairseq_adam=FairseqAdam,
   adamw=torch.optim.AdamW,