fix: do not run 4 times and log new params

2026-06-02 12:00:12 +02:00
parent 2556c17d88
commit 150c2a7253
3 changed files with 360 additions and 347 deletions
@@ -49,6 +49,8 @@ def build_run_name(args):
    val_args = args.get("validation", {})

    model_name = meta_args.get("model_name", "model")
+    representation_type = meta_args.get("representation_type")
+    head_type = meta_args.get("head_type")
    patch_size = mask_args.get("patch_size", meta_args.get("patch_size"))
    crop_size = data_args.get("crop_size", meta_args.get("crop_size"))
    batch_size = data_args.get("batch_size")
@@ -67,6 +69,8 @@ def build_run_name(args):
    add("p", patch_size)
    add("c", crop_size)
    add("bs", batch_size)
+    add("rep", representation_type)
+    add("head", head_type)
    parts.append(str(optimizer).lower())
    add("lr", lr)
    add("wd", weight_decay)