funasr/build_utils/build_optimizer.py
@@ -23,4 +23,6 @@ if optim_class is None: raise ValueError(f"must be one of {list(optim_classes)}: {args.optim}") optimizer = optim_class(model.parameters(), **args.optim_conf) return optimizer optimizers = [optimizer] return optimizers
