shixian.shi
2024-01-11 70379713923e9938dcfcb3791b17f7b469233432
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,