diff --git a/configs/supervised_wilds_vith14_ep300-lp.yaml b/configs/supervised_wilds_vith14_ep300-lp.yaml
index 9bfeb19..b228a88 100644
--- a/configs/supervised_wilds_vith14_ep300-lp.yaml
+++ b/configs/supervised_wilds_vith14_ep300-lp.yaml
@@ -8,7 +8,7 @@ meta:
   num_classes: 182
 
 data:
-  batch_size: 16384
+  batch_size: 256
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true
@@ -52,5 +52,5 @@ validation:
   eval_every: 1
 
 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.01-wd0.0005/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.01-wd0.0005/
   write_tag: linear_probe-lars-lr0.01-wd0.0005
diff --git a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0.yaml b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0.yaml
index 5e5225a..382e148 100644
--- a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0.yaml
+++ b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0.yaml
@@ -8,7 +8,7 @@ meta:
   num_classes: 182
 
 data:
-  batch_size: 16384
+  batch_size: 256
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true
@@ -52,5 +52,5 @@ validation:
   eval_every: 1
 
 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.001-wd0.0/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.001-wd0.0/
   write_tag: linear_probe-lars-lr0.001-wd0.0
diff --git a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0005.yaml b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0005.yaml
index c9c9807..d59fb85 100644
--- a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0005.yaml
+++ b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0005.yaml
@@ -8,7 +8,7 @@ meta:
   num_classes: 182
 
 data:
-  batch_size: 16384
+  batch_size: 256
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true
@@ -52,5 +52,5 @@ validation:
   eval_every: 1
 
 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.001-wd0.0005/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.001-wd0.0005/
   write_tag: linear_probe-lars-lr0.001-wd0.0005
diff --git a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.01-wd0.0.yaml b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.01-wd0.0.yaml
index dd00493..f106943 100644
--- a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.01-wd0.0.yaml
+++ b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.01-wd0.0.yaml
@@ -8,7 +8,7 @@ meta:
   num_classes: 182
 
 data:
-  batch_size: 16384
+  batch_size: 256
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true
@@ -52,5 +52,5 @@ validation:
   eval_every: 1
 
 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.01-wd0.0/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.01-wd0.0/
   write_tag: linear_probe-lars-lr0.01-wd0.0
diff --git a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0.yaml b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0.yaml
index 2d9b08e..fac42ff 100644
--- a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0.yaml
+++ b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0.yaml
@@ -8,7 +8,7 @@ meta:
   num_classes: 182
 
 data:
-  batch_size: 16384
+  batch_size: 256
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true
@@ -52,5 +52,5 @@ validation:
   eval_every: 1
 
 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.05-wd0.0/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.05-wd0.0/
   write_tag: linear_probe-lars-lr0.05-wd0.0
diff --git a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0005.yaml b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0005.yaml
index c9f9879..f285673 100644
--- a/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0005.yaml
+++ b/configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0005.yaml
@@ -8,7 +8,7 @@ meta:
   num_classes: 182
 
 data:
-  batch_size: 16384
+  batch_size: 256
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true
@@ -52,5 +52,5 @@ validation:
   eval_every: 1
 
 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.05-wd0.0005/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.05-wd0.0005/
   write_tag: linear_probe-lars-lr0.05-wd0.0005