diff --git a/configs/wilds_vitb16-448_ep300.yaml b/configs/wilds_vitb16-448_ep300.yaml
index d655b66..f99e3c1 100644
--- a/configs/wilds_vitb16-448_ep300.yaml
+++ b/configs/wilds_vitb16-448_ep300.yaml
@@ -1,5 +1,5 @@
 data:
-  batch_size: 128
+  batch_size: 16
   color_jitter_strength: 0.0
   crop_scale:
     - 1.0
@@ -43,7 +43,7 @@ optimization:
   ema:
     - 0.996
     - 1.0
-  epochs: 300
+  epochs: 600
   final_lr: 1.0e-06
   final_weight_decay: 0.4
   ipe_scale: 1.0