updated configs with more SPS

dunnolab · Jul 17, 2024 · 94759d3 · 94759d3
1 parent a2e8f67
commit 94759d3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/configs/meta.yaml b/configs/meta.yaml
@@ -4,9 +4,9 @@ benchmark_id: "high-1m"
 total_timesteps: 10_000_000_000
 num_envs: 16384
 num_steps_per_env: 12800
-num_steps_per_update: 256
+num_steps_per_update: g256
 update_epochs: 1
-num_minibatches: 32
+num_minibatches: 16
 eval_num_envs: 16384
 eval_num_episodes: 25
 train_seed: 5
diff --git a/configs/single.yaml b/configs/single.yaml
@@ -6,6 +6,6 @@ total_timesteps: 1_000_000_000
 num_envs: 16384
 num_steps: 256
 update_epochs: 1
-num_minibatches: 8
+num_minibatches: 16
 eval_episodes: 512
 train_seed: 5