small updates

Denys88 · Nov 24, 2023 · 7f3398c · 7f3398c
1 parent d07b29c
commit 7f3398c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rl_games/configs/smac/v2/terran_5_v_5.yaml b/rl_games/configs/smac/v2/terran_5_v_5.yaml
@@ -41,7 +41,7 @@ params:
     clip_value: False
     num_actors: 16
     horizon_length: 256
-    minibatch_size: 2560 # 5 * 512
+    minibatch_size: 10240 # 5 * 512
     mini_epochs: 4
     critic_coef: 1
     lr_schedule: linear

diff --git a/rl_games/envs/smac_v2_env.py b/rl_games/envs/smac_v2_env.py
@@ -14,7 +14,7 @@ def __init__(self, name="3m",  **kwargs):
         self.use_central_value = kwargs.pop('central_value', True)
         self.concat_infos = True
         self.random_invalid_step = kwargs.pop('random_invalid_step', False)
-        self.replay_save_freq = kwargs.pop('replay_save_freq', 10000)
+        self.replay_save_freq = kwargs.pop('replay_save_freq', 100)
         self.apply_agent_ids = kwargs.pop('apply_agent_ids', True)
         with open(self.path, 'r') as stream:
             config = yaml.safe_load(stream)