diff --git a/configs/grids/lp_grid.yaml b/configs/grids/lp_grid.yaml
index 23e9c70..70ec4ee 100644
--- a/configs/grids/lp_grid.yaml
+++ b/configs/grids/lp_grid.yaml
@@ -1,4 +1,4 @@
-base_config: configs/supervised_wilds_vitg16_ep300-lp.yaml
+base_config: configs/supervised_wilds_vith16-448_ep300-lp.yaml
 
 constants:
   logging.write_tag: linear_probe
diff --git a/configs/supervised_wilds_vith16-448_ep300-lp.yaml b/configs/supervised_wilds_vith16-448_ep300-lp.yaml
index bbb42c5..098996f 100644
--- a/configs/supervised_wilds_vith16-448_ep300-lp.yaml
+++ b/configs/supervised_wilds_vith16-448_ep300-lp.yaml
@@ -1,10 +1,9 @@
 meta:
   model_name: vit_huge
-  tag: in1k
   embed_dim: 1280
   load_checkpoint: true
-  checkpoint_folder: experiment_logs/imagenet-vith16.448/
-  read_checkpoint: IN1K-vit.h.16-448px-300e.pth.tar
+  checkpoint_folder: experiment_logs/vith16-bs.2048-448/
+  read_checkpoint: jepa-ep50.pth.tar
   use_bfloat16: true
   num_classes: 182
   representation_type: last_avgpool