1
Fork 0

feat(config): add checkpoint frequency parameter

Add checkpoint_frequency to PPOArgs to enable periodic checkpoint
saving during training. Default: save every 100 iterations.
This commit is contained in:
Tibo De Peuter 2026-03-31 19:49:12 +00:00
parent 3ce107a560
commit bb0bb94f60

View file

@ -26,7 +26,7 @@ class PPOArgs:
wandb_project_name: str = "PPO-Modularity"
# the entity (team) of wandb's project
wandb_entity: str | None = None
wandb_entity: str | None = "tdpeuter-ghent-university"
# whether to capture videos of the agent performances (check out `videos` folder)
capture_video: bool = False
@ -34,6 +34,9 @@ class PPOArgs:
# whether to save model into the `runs/{run_name}` folder
save_model: bool = True
# checkpoint frequency (in iterations, 0 = no intermediate checkpoints)
checkpoint_frequency: int = 100
# whether to upload the saved model to huggingface
upload_model: bool = False