Yabin Li
2023-10-16 c38e4f39f6b52c278294a11c9e06c260ea8e4ecd
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