| """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() | |