haoneng.lhn
2023-08-10 5134dfea33636a301d5f46dd86e17531536e2934
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