shixian.shi
2024-01-10 f79d31d86010a95ad45efe6cac5e5d0f95e4f35a
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,