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

new grid

Browse Source
This commit is contained in:
YannAhlgrim
2026-05-25 11:33:21 +02:00
parent 9c82ba523b
commit ace8c4fd97
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/grids/lp_grid.yaml
+8 -8
View File
@@ -4,15 +4,15 @@ constants:
logging.write_tag: linear_probe logging.write_tag: linear_probe
grid: grid:
optimization.optimizer: [adamw, lars, sgd] optimization.optimizer: [adamw]
optimization.lr: [0.01, 0.05] optimization.lr: [0.01]
optimization.weight_decay: [5.0e-4] optimization.weight_decay: [5.0e-4]
optimization.momentum: [0.9, 0.5] optimization.momentum: [0.9, 0.99]
optimization.lr_schedule: [cosine, step] optimization.lr_schedule: [cosine]
data.batch_size: [32, 64, 512] data.batch_size: [128, 256]
meta.probe_type: [linear, mlp] meta.probe_type: [mlp, linear]
meta.mlp_hidden_dim: [1024] meta.mlp_hidden_dim: [512, 1024, 2048]
meta.dropout: [0.0] meta.dropout: [0.0, 0.1, 0.3]
launch: launch:
folder: submitit_logs/ folder: submitit_logs/