feat: configure checkpoints saving
This commit is contained in:
parent
4d0f729aee
commit
37a4b59e04
10 changed files with 65 additions and 36 deletions
|
|
@ -6,6 +6,8 @@ wandb_project_name: "PPO-Modularity"
|
|||
wandb_entity: "SEL3-2026-Groep-4"
|
||||
capture_video: false
|
||||
save_model: true
|
||||
save_checkpoints: true
|
||||
checkpoint_frequency: 100
|
||||
upload_model: false
|
||||
upload_final_model: false
|
||||
upload_checkpoints: false
|
||||
hf_entity: ""
|
||||
|
|
|
|||
9
configs/logging/hpc.yaml
Normal file
9
configs/logging/hpc.yaml
Normal file
|
|
@ -0,0 +1,9 @@
|
|||
track: true
|
||||
wandb_project_name: "hpc-default"
|
||||
wandb_entity: "SEL3-2026-Groep-4"
|
||||
save_model: true
|
||||
save_checkpoints: true
|
||||
upload_final_model: true
|
||||
upload_checkpoints: true
|
||||
checkpoint_frequency: 100
|
||||
hf_entity: ""
|
||||
|
|
@ -2,10 +2,12 @@
|
|||
# For production/cloud experiments with weights synced.
|
||||
|
||||
track: true
|
||||
wandb_project_name: "PPO-Modularity"
|
||||
wandb_project_name: "default-project"
|
||||
wandb_entity: "SEL3-2026-Groep-4"
|
||||
capture_video: false
|
||||
save_model: true
|
||||
save_checkpoints: true
|
||||
checkpoint_frequency: 100
|
||||
upload_model: false
|
||||
upload_final_model: true
|
||||
upload_checkpoints: false
|
||||
hf_entity: ""
|
||||
|
|
|
|||
Reference in a new issue