From 001babffbdaae3fded7d397f8ca373845f87366a Mon Sep 17 00:00:00 2001
From: Yann Ahlgrim <yann@yann-x1carbon.ahlgrim.net>
Date: Fri, 26 Jun 2026 14:29:32 +0200
Subject: [PATCH] run for imagenet vith ep300 model

---
 configs/supervised_wilds_vith14_ep300-lp.yaml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/configs/supervised_wilds_vith14_ep300-lp.yaml b/configs/supervised_wilds_vith14_ep300-lp.yaml
index 6e266ff..158eb78 100644
--- a/configs/supervised_wilds_vith14_ep300-lp.yaml
+++ b/configs/supervised_wilds_vith14_ep300-lp.yaml
@@ -1,9 +1,10 @@
 meta:
   model_name: vit_huge
+  tag: in1k
   embed_dim: 1280
   load_checkpoint: true
-  checkpoint_folder: experiment_logs/vith14.224-bs.128-ep.300/
-  read_checkpoint: jepa-ep300.pth.tar
+  checkpoint_folder: experiment_logs/imagenet-vith16.448/
+  read_checkpoint: IN1K-vit.h.14-300e.pth.tar
   use_bfloat16: true
   num_classes: 182
   representation_type: last_avgpool