Get sb3 checkpoints saving during training

2023-11-16 14:37:37 +00:00
parent 4cc7ba1522
commit 829500a60f
5 changed files with 36 additions and 19 deletions
--- a/src/primaite/config/_package_data/example_config.yaml
+++ b/src/primaite/config/_package_data/example_config.yaml
@@ -2,7 +2,7 @@ training_config:
  rl_framework: SB3
  rl_algorithm: PPO
  seed: 333
-  n_learn_steps: 2560
+  n_learn_episodes: 25
  n_eval_episodes: 5
  max_steps_per_episode: 128
  deterministic_eval: false
@@ -10,6 +10,10 @@ training_config:
  agent_references:
    - defender

+io_settings:
+  save_checkpoints: true
+  checkpoint_interval: 5
+

 game_config:
  ports: