bug: missing fields

2026-05-27 17:08:58 +02:00
parent 87b0c16c24
commit a9599c7220
2 changed files with 6 additions and 0 deletions
@@ -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

@@ -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,
            },