From 0fcab1d3c7e701c6d5c519c7e81cd877a3df7590 Mon Sep 17 00:00:00 2001 From: "ahyanc25@reutlingen-university.de" Date: Wed, 20 May 2026 15:41:45 +0200 Subject: [PATCH] pred depth --- configs/wilds_vitb16-448_ep600.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/wilds_vitb16-448_ep600.yaml b/configs/wilds_vitb16-448_ep600.yaml index e4c88e5..27f06ec 100644 --- a/configs/wilds_vitb16-448_ep600.yaml +++ b/configs/wilds_vitb16-448_ep600.yaml @@ -35,7 +35,7 @@ meta: load_checkpoint: false model_name: vit_base embed_dim: 768 - pred_depth: 12 + pred_depth: 6 pred_emb_dim: 384 read_checkpoint: null use_bfloat16: true