Yabin Li
2023-06-30 057d349911fee0a4d7a3560e50752272ad2fe59d
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