diff --git a/configs/supervised_wilds_vith14_ep300-lp.yaml b/configs/supervised_wilds_vith14_ep300-lp.yaml index 158eb78..7f032f7 100644 --- a/configs/supervised_wilds_vith14_ep300-lp.yaml +++ b/configs/supervised_wilds_vith14_ep300-lp.yaml @@ -1,10 +1,10 @@ meta: model_name: vit_huge - tag: in1k + tag: in22k embed_dim: 1280 load_checkpoint: true checkpoint_folder: experiment_logs/imagenet-vith16.448/ - read_checkpoint: IN1K-vit.h.14-300e.pth.tar + read_checkpoint: IN22K-vit.h.14-900e.pth.tar use_bfloat16: true num_classes: 182 representation_type: last_avgpool