bug: missing fields

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