reduce batch size to fit on 2 A100

ymahlau · Mar 28, 2024 · 2963a40 · 2963a40
1 parent bd26868
commit 2963a40
Show file tree

Hide file tree

Showing 10 changed files with 10 additions and 10 deletions.
diff --git a/config/cfg_4dc11_0.yaml b/config/cfg_4dc11_0.yaml
@@ -145,7 +145,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_1.yaml b/config/cfg_4dc11_1.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_2.yaml b/config/cfg_4dc11_2.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_3.yaml b/config/cfg_4dc11_3.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_4.yaml b/config/cfg_4dc11_4.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_proxy_0.yaml b/config/cfg_4dc11_proxy_0.yaml
@@ -145,7 +145,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_proxy_1.yaml b/config/cfg_4dc11_proxy_1.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_proxy_2.yaml b/config/cfg_4dc11_proxy_2.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_proxy_3.yaml b/config/cfg_4dc11_proxy_3.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2

diff --git a/config/cfg_4dc11_proxy_4.yaml b/config/cfg_4dc11_proxy_4.yaml
@@ -161,7 +161,7 @@ data:
       updater_bucket_size: 1000
       wandb_mode: offline
       worker_episode_bucket_size: 20
-  max_batch_size: 12000
+  max_batch_size: 8192
   max_cpu_evaluator: 2
   max_cpu_inference_server: 5
   max_cpu_log_dist_save_collect: 2