File size: 635 Bytes
9613b5e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | 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']}")
|