diff --git a/configs/wilds_vith16-448_ep300.yaml b/configs/wilds_vith16-448_ep300.yaml
index 6200821..8f09488 100644
--- a/configs/wilds_vith16-448_ep300.yaml
+++ b/configs/wilds_vith16-448_ep300.yaml
@@ -1,5 +1,5 @@
 data:
-  batch_size: 16
+  batch_size: 128
   color_jitter_strength: 0.0
   crop_scale:
     - 1.0
@@ -40,7 +40,7 @@ meta:
   use_bfloat16: true
   use_gradient_checkpointing: true
 optimization:
-  gradient_accumulation_steps: 21
+  gradient_accumulation_steps: 2
   ema:
     - 0.996
     - 1.0