andyweiqiu
2023-09-07 72a88f22f00c4567503adf852dd2f3f2938ecf7c
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