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

vith 16 run

Browse Source
This commit is contained in:
YannAhlgrim
2026-07-01 08:45:12 +02:00
parent b46d86e517
commit 70b61747bf
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/grids/lp_grid.yaml
+1 -1
View File
@@ -1,4 +1,4 @@
base_config: configs/supervised_wilds_vitg16_ep300-lp.yaml base_config: configs/supervised_wilds_vith16-448_ep300-lp.yaml
constants: constants:
logging.write_tag: linear_probe logging.write_tag: linear_probe
configs/supervised_wilds_vith16-448_ep300-lp.yaml
+2 -3
View File
@@ -1,10 +1,9 @@
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/imagenet-vith16.448/ checkpoint_folder: experiment_logs/vith16-bs.2048-448/
read_checkpoint: IN1K-vit.h.16-448px-300e.pth.tar read_checkpoint: jepa-ep50.pth.tar
use_bfloat16: true use_bfloat16: true
num_classes: 182 num_classes: 182
representation_type: last_avgpool representation_type: last_avgpool