| from pathlib import Path | |
| import sys | |
| ROOT = Path(__file__).resolve().parents[1] | |
| sys.path.insert(0, str(ROOT)) | |
| from model.atmosphericda_dmodel import QGConfig, load_yaml, train_model | |
| if __name__ == "__main__": | |
| config = load_yaml(ROOT / "conf/config.yaml") | |
| config["training"]["format_version"] = config["project"]["format_version"] | |
| history = train_model( | |
| str(ROOT / config["paths"]["data"]), | |
| str(ROOT / config["paths"]["checkpoint"]), | |
| str(ROOT / config["paths"]["training_metrics"]), | |
| QGConfig(**config["qg"]), config["model"], config["training"], | |
| ) | |
| print(f"loss={history['loss']}") | |