From f152e8a1e9164594e8fdac1c86522f08bdccc68b Mon Sep 17 00:00:00 2001
From: Daphne Cornelisse <cor.daphne@gmail.com>
Date: Tue, 14 Jan 2025 10:04:54 -0500
Subject: [PATCH] Save consts

---
 examples/experiments/ippo_ff_p1_self_play.yaml | 12 ++++++------
 src/consts.hpp                                 |  2 +-
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/examples/experiments/ippo_ff_p1_self_play.yaml b/examples/experiments/ippo_ff_p1_self_play.yaml
index ba5d102c..0c947fb0 100644
--- a/examples/experiments/ippo_ff_p1_self_play.yaml
+++ b/examples/experiments/ippo_ff_p1_self_play.yaml
@@ -2,13 +2,13 @@ mode: "train"
 use_rnn: false
 eval_model_path: null
 baseline: false
-data_dir: "data/processed/training"
+data_dir: "data/processed/examples"
 
 environment: # Overrides default environment configs (see pygpudrive/env/config.py)
   name: "gpudrive"
-  num_worlds: 100 # Number of parallel environments
-  k_unique_scenes: 100 # Number of unique scenes to sample from
-  max_controlled_agents: 128 # Maximum number of agents controlled by the model. Make sure this aligns with the variable kMaxAgentCount in src/consts.hpp
+  num_worlds: 50 # Number of parallel environments
+  k_unique_scenes: 1 # Number of unique scenes to sample from
+  max_controlled_agents: 64 # Maximum number of agents controlled by the model. Make sure this aligns with the variable kMaxAgentCount in src/consts.hpp
   ego_state: true
   road_map_obs: true
   partner_obs: true
@@ -43,7 +43,7 @@ train:
   compile_mode: "reduce-overhead"
 
   # # # Data sampling # # #
-  resample_scenes: true
+  resample_scenes: false
   resample_criterion: "global_step"
   resample_dataset_size: 1000 # Number of unique scenes to sample from
   resample_interval: 1_000_000
@@ -65,7 +65,7 @@ train:
   clip_coef: 0.2
   clip_vloss: false
   vf_clip_coef: 0.2
-  ent_coef: 0.0001
+  ent_coef: 0.001
   vf_coef: 0.5
   max_grad_norm: 0.5
   target_kl: null
diff --git a/src/consts.hpp b/src/consts.hpp
index 98822e5a..90be2638 100644
--- a/src/consts.hpp
+++ b/src/consts.hpp
@@ -8,7 +8,7 @@ namespace gpudrive {
 
 namespace consts {
 
-inline constexpr madrona::CountT kMaxAgentCount = 128;
+inline constexpr madrona::CountT kMaxAgentCount = 64;
 inline constexpr madrona::CountT kMaxRoadEntityCount = 10000;
 inline constexpr madrona::CountT kMaxAgentMapObservationsCount = 200;