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