shixian.shi
2024-01-12 09a28d19df5854bdd4bd4d3a05dcb6f502ec6b07
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,