1
Fork 0

feat(hpc): support run_dir staging and update docs

- Add run_dir and checkpoint_frequency to PPOArgs
- Update train.py to use run_dir for SummaryBoard, model saving, and loss plots
- Create configs/production_training.yaml for HPC production runs
- Update HPC.md with run_dir staging strategy details
This commit is contained in:
Tibo De Peuter 2026-04-04 18:59:05 +02:00
parent 34a887cd58
commit 92db8c2591
5 changed files with 42 additions and 50 deletions

View file

@ -13,6 +13,12 @@ class PPOArgs:
# the name of this experiment
exp_name: str = "brittle_star_ppo"
# the directory to save the experiment results
run_dir: str | None = None
# how often to save checkpoints (0 to disable)
checkpoint_frequency: int = 0
# seed of the experiment
seed: int = 1