File size: 659 Bytes
1f88cea | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | #!/usr/bin/env python3
"""Run an official BoltzGen Hydra training configuration."""
from __future__ import annotations
import os
import sys
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
MODEL_DIR = ROOT / "model"
sys.path.insert(0, str(MODEL_DIR))
os.environ["PYTHONPATH"] = os.pathsep.join(
[str(MODEL_DIR), os.environ.get("PYTHONPATH", "")]
).rstrip(os.pathsep)
from boltzgen.resources.main import main # noqa: E402
def run() -> None:
if len(sys.argv) < 2:
raise SystemExit("Usage: python scripts/train.py CONFIG.yaml [key=value ...]")
main(sys.argv[1], sys.argv[2:])
if __name__ == "__main__":
run()
|