File size: 560 Bytes
6f3c6ef | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | """Train the local model from scratch on virtual FV3GFS-contract data."""
import argparse
from local_pipeline import load_config, train
def main() -> None:
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--config", default="conf/config.yaml")
parser.add_argument(
"--finetune",
help="Optional checkpoint path. Without this argument, training starts from scratch.",
)
args = parser.parse_args()
train(load_config(args.config), finetune=args.finetune)
if __name__ == "__main__":
main()
|