amend

pytorch · Apr 22, 2024 · 12db41f · 12db41f
1 parent a9e1cb0
commit 12db41f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/unittest/linux_examples/scripts/run_test.sh b/.github/unittest/linux_examples/scripts/run_test.sh
@@ -170,12 +170,12 @@ python .github/unittest/helpers/coverage_run_parallel.py sota-implementations/dr
   collector.total_frames=200 \
   collector.init_random_frames=10 \
   collector.frames_per_batch=200 \
-  replay_buffer.batch_size=10 \
   env.n_parallel_envs=4 \
   optimization.optim_steps_per_batch=1 \
   logger.video=True \
   logger.backend=csv \
   replay_buffer.buffer_size=120 \
+  replay_buffer.batch_length=12 \
   networks.rssm_hidden_dim=17
 python .github/unittest/helpers/coverage_run_parallel.py sota-implementations/td3/td3.py \
   collector.total_frames=48 \
@@ -223,12 +223,12 @@ python .github/unittest/helpers/coverage_run_parallel.py sota-implementations/dr
   collector.total_frames=200 \
   collector.init_random_frames=10 \
   collector.frames_per_batch=200 \
-  replay_buffer.batch_size=10 \
   env.n_parallel_envs=1 \
   optimization.optim_steps_per_batch=1 \
   logger.backend=csv \
   logger.video=True \
   replay_buffer.buffer_size=120 \
+  replay_buffer.batch_length=12 \
   networks.rssm_hidden_dim=17
 python .github/unittest/helpers/coverage_run_parallel.py sota-implementations/ddpg/ddpg.py \
   collector.total_frames=48 \