From cf7b1bad7522fd6541055d9d201e2c74ce08d7b9 Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Wed, 26 Jun 2024 13:43:11 -0400 Subject: [PATCH] Increase batch size for debug mode --- config.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.yaml b/config.yaml index 9645b4e..cfbbf44 100644 --- a/config.yaml +++ b/config.yaml @@ -22,7 +22,7 @@ debug: env_batch_size: 4 env_pool: True zero_copy: False - batch_size: 4 + batch_size: 128 minibatch_size: 4 batch_rows: 4 bptt_horizon: 2