YannAhlgrim/wilds-ijepa
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add tag to folder name + change grid config yaml

Browse Source
This commit is contained in:
Yann Ahlgrim
2026-06-19 10:18:16 +02:00
parent c7d080f2cd
commit 2308540971
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/grids/lp_grid.yaml
+1 -1
View File
@@ -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: constants:
logging.write_tag: linear_probe logging.write_tag: linear_probe
configs/supervised_wilds_vith16-448_ep300-lp.yaml
+1 -1
View File
@@ -1,5 +1,6 @@
meta: meta:
model_name: vit_huge model_name: vit_huge
tag: in1k
embed_dim: 1280 embed_dim: 1280
load_checkpoint: true load_checkpoint: true
checkpoint_folder: experiment_logs/imagenet-vith16.448/ checkpoint_folder: experiment_logs/imagenet-vith16.448/
@@ -56,6 +57,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
run_name: in1k-vith16-448
write_tag: linear_probe write_tag: linear_probe
auto_folder: true auto_folder: true
src/utils/logging.py
+3
View File
@@ -49,6 +49,7 @@ def build_run_name(args):
val_args = args.get("validation", {}) val_args = args.get("validation", {})
model_name = meta_args.get("model_name", "model") model_name = meta_args.get("model_name", "model")
tag = meta_args.get("tag")
representation_type = meta_args.get("representation_type") representation_type = meta_args.get("representation_type")
head_type = meta_args.get("head_type") head_type = meta_args.get("head_type")
patch_size = mask_args.get("patch_size", meta_args.get("patch_size")) 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("p", patch_size)
add("c", crop_size) add("c", crop_size)
if tag:
parts.append(str(tag))
add("bs", batch_size) add("bs", batch_size)
add("rep", representation_type) add("rep", representation_type)
add("head", head_type) add("head", head_type)