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

grid for vit b ep 600

Browse Source
This commit is contained in:
YannAhlgrim
2026-06-15 13:52:07 +02:00
parent 8b714927ab
commit 1ceaf09a13
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/grids/lp_grid.yaml
+3 -3
View File
@@ -8,8 +8,8 @@ grid:
optimization.lr: [0.01, 0.05, 0.001] optimization.lr: [0.01, 0.05, 0.001]
optimization.weight_decay: [5.0e-4, 0] optimization.weight_decay: [5.0e-4, 0]
optimization.momentum: [0.9, 0.5] optimization.momentum: [0.9, 0.5]
optimization.lr_schedule: [cosine, step] optimization.lr_schedule: [cosine]
data.batch_size: [64, 128, 256] data.batch_size: [256, 512, 1024]
meta.representation_type: [last_avgpool, last4_avgpool_concat] meta.representation_type: [last_avgpool, last4_avgpool_concat]
meta.head_type: [linear, bn_linear] meta.head_type: [linear, bn_linear]
@@ -18,4 +18,4 @@ launch:
partition: gpu1 partition: gpu1
nodes: 1 nodes: 1
tasks_per_node: 1 tasks_per_node: 1
time: 180 time: 600
configs/supervised_wilds_vitb16-448_ep600-lp.yaml
+1 -1
View File
@@ -3,7 +3,7 @@ meta:
embed_dim: 768 embed_dim: 768
load_checkpoint: true load_checkpoint: true
checkpoint_folder: experiment_logs/vitb16.448-bs.16-ep.600/ checkpoint_folder: experiment_logs/vitb16.448-bs.16-ep.600/
read_checkpoint: jepa-ep150.pth.tar read_checkpoint: jepa-ep600.pth.tar
use_bfloat16: true use_bfloat16: true
num_classes: 182 num_classes: 182
representation_type: last_avgpool representation_type: last_avgpool