vith 16 run

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
+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:
logging.write_tag: linear_probe
@@ -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