OneScience-Group/AtmosphericDA-DModel

1from pathlib import Path2import sys3 4ROOT = Path(__file__).resolve().parents[1]5sys.path.insert(0, str(ROOT))6 7from model.atmosphericda_dmodel import QGConfig, load_yaml, train_model8 9 10if __name__ == "__main__":11    config = load_yaml(ROOT / "conf/config.yaml")12    config["training"]["format_version"] = config["project"]["format_version"]13    history = train_model(14        str(ROOT / config["paths"]["data"]),15        str(ROOT / config["paths"]["checkpoint"]),16        str(ROOT / config["paths"]["training_metrics"]),17        QGConfig(**config["qg"]), config["model"], config["training"],18    )19    print(f"loss={history['loss']}")20