YannAhlgrim/wilds-ijepa
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

run for imagenet vith ep300 model

Browse Source
This commit is contained in:
Yann Ahlgrim
2026-06-26 14:29:32 +02:00
parent 2332402849
commit 001babffbd
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/supervised_wilds_vith14_ep300-lp.yaml
+3 -2
View File
@@ -1,9 +1,10 @@
meta: meta:
model_name: vit_huge model_name: vit_huge
tag: in1k
embed_dim: 1280 embed_dim: 1280
load_checkpoint: true load_checkpoint: true
checkpoint_folder: experiment_logs/vith14.224-bs.128-ep.300/ checkpoint_folder: experiment_logs/imagenet-vith16.448/
read_checkpoint: jepa-ep300.pth.tar read_checkpoint: IN1K-vit.h.14-300e.pth.tar
use_bfloat16: true use_bfloat16: true
num_classes: 182 num_classes: 182
representation_type: last_avgpool representation_type: last_avgpool