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

bug: missing fields

Browse Source
This commit is contained in:
YannAhlgrim
2026-05-27 17:08:58 +02:00
parent 87b0c16c24
commit a9599c7220
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/eval_wilds_vith14.yaml
+3
View File
@@ -6,6 +6,9 @@ meta:
patch_size: 14 patch_size: 14
crop_size: 224 crop_size: 224
use_bfloat16: true use_bfloat16: true
probe_type: linear
mlp_hidden_dim:
dropout: 0.0
checkpoint_folder: experiment_logs/supervised/ checkpoint_folder: experiment_logs/supervised/
read_checkpoint: linear_probe-best.pth.tar read_checkpoint: linear_probe-best.pth.tar
src/train_supervised.py
+3
View File
@@ -489,6 +489,9 @@ def main(args, resume_preempt=False):
"patch_size": mk_args.get("patch_size", m_args.get("patch_size", 16)), "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)), "crop_size": d_args.get("crop_size", m_args.get("crop_size", 224)),
"use_bfloat16": m_args.get("use_bfloat16", True), "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, "checkpoint_path": best_path,
"force_single_process": True, "force_single_process": True,
}, },