diff --git a/configs/eval_wilds_vith14.yaml b/configs/eval_wilds_vith14.yaml
index 7c2f8b4..b7ce3f4 100644
--- a/configs/eval_wilds_vith14.yaml
+++ b/configs/eval_wilds_vith14.yaml
@@ -6,6 +6,9 @@ meta:
   patch_size: 14
   crop_size: 224
   use_bfloat16: true
+  probe_type: linear
+  mlp_hidden_dim:
+  dropout: 0.0
   checkpoint_folder: experiment_logs/supervised/
   read_checkpoint: linear_probe-best.pth.tar
 
diff --git a/src/train_supervised.py b/src/train_supervised.py
index 4e5f34b..3a3a5a0 100644
--- a/src/train_supervised.py
+++ b/src/train_supervised.py
@@ -489,6 +489,9 @@ def main(args, resume_preempt=False):
                 "patch_size": mk_args.get("patch_size", m_args.get("patch_size", 16)),
                 "crop_size": d_args.get("crop_size", m_args.get("crop_size", 224)),
                 "use_bfloat16": m_args.get("use_bfloat16", True),
+                "probe_type": m_args.get("probe_type", "linear"),
+                "mlp_hidden_dim": m_args.get("mlp_hidden_dim"),
+                "dropout": m_args.get("dropout", 0.0),
                 "checkpoint_path": best_path,
                 "force_single_process": True,
             },