diff --git a/configs/supervised_wilds_vitb16-448_ep600-lp.yaml b/configs/supervised_wilds_vitb16-448_ep600-lp.yaml
index 93baf2a..c775619 100644
--- a/configs/supervised_wilds_vitb16-448_ep600-lp.yaml
+++ b/configs/supervised_wilds_vitb16-448_ep600-lp.yaml
@@ -10,7 +10,7 @@ meta:
   head_type: linear
 
 data:
-  batch_size: 256
+  batch_size: 1024
   root_path: ./wilds_data
   num_workers: 10
   pin_mem: true