fix(train): integrate YAML config
This commit is contained in:
parent
2001a92e75
commit
e160a55d95
3 changed files with 10 additions and 2 deletions
|
|
@ -26,7 +26,7 @@ class PPOArgs:
|
|||
wandb_project_name: str = "PPO-Modularity"
|
||||
|
||||
# the entity (team) of wandb's project
|
||||
wandb_entity: str | None = "tdpeuter-ghent-university"
|
||||
wandb_entity: str | None = None
|
||||
|
||||
# whether to capture videos of the agent performances (check out `videos` folder)
|
||||
capture_video: bool = False
|
||||
|
|
|
|||
Reference in a new issue