add tag to folder name + change grid config yaml

configs/grids/lp_grid.yaml

@@ -1,4 +1,4 @@
-base_config: configs/supervised_wilds_vitb16-448_ep600-lp.yaml
+base_config: configs/supervised_wilds_vith16-448_ep300-lp.yaml
 
 constants:
   logging.write_tag: linear_probe

configs/supervised_wilds_vith16-448_ep300-lp.yaml

@@ -1,5 +1,6 @@
 meta:
   model_name: vit_huge
+  tag: in1k
   embed_dim: 1280
   load_checkpoint: true
   checkpoint_folder: experiment_logs/imagenet-vith16.448/
@@ -56,6 +57,5 @@ validation:
   eval_every: 1
 
 logging:
-  run_name: in1k-vith16-448
   write_tag: linear_probe
   auto_folder: true

src/utils/logging.py

@@ -49,6 +49,7 @@ def build_run_name(args):
     val_args = args.get("validation", {})
 
     model_name = meta_args.get("model_name", "model")
+    tag = meta_args.get("tag")
     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"))
@@ -68,6 +69,8 @@ def build_run_name(args):
 
     add("p", patch_size)
     add("c", crop_size)
+    if tag:
+        parts.append(str(tag))
     add("bs", batch_size)
     add("rep", representation_type)
     add("head", head_type)