diff --git a/configs/meta.yaml b/configs/meta.yaml index 12795d8..c12edd9 100644 --- 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 \ No newline at end of file diff --git a/configs/single.yaml b/configs/single.yaml index 11d45fa..e106d04 100644 --- 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 \ No newline at end of file