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

try new params

Browse Source
This commit is contained in:
YannAhlgrim
2026-05-21 16:05:26 +02:00
parent 29e0d4d64f
commit 56cdf38d8c
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/eval_wilds_vith14.yaml
+3 -3
View File
@@ -6,8 +6,8 @@ meta:
patch_size: 14
crop_size: 224
use_bfloat16: true
checkpoint_folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.001-wd0.0005/
read_checkpoint: linear_probe-lars-lr0.001-wd0.0005-ep50.pth.tar
checkpoint_folder: experiment_logs/supervised/
read_checkpoint: linear_probe.pth.tar
data:
batch_size: 128
@@ -18,5 +18,5 @@ data:
download: true
logging:
folder: experiment_logs/eval-wilds/vith14.224-bs.128-ep.300-ft/
folder: experiment_logs/eval-wilds/vith14-ep300-adamw-ep.50-lr0.01-wd0.0005/
write_tag: iwildcam_test
configs/supervised_wilds_vith14_ep300-lp.yaml
+3 -3
View File
@@ -24,7 +24,7 @@ mask:
patch_size: 14
optimization:
optimizer: lars # 'adamw', 'sgd', or 'lars'
optimizer: adamw # 'adamw', 'sgd', or 'lars'
freeze_weights: true # true for linear probing, false for full fine-tuning
epochs: 50
lr: 0.01
@@ -55,5 +55,5 @@ validation:
eval_every: 1
logging:
folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.01-wd0.0005/
write_tag: linear_probe-lars-lr0.01-wd0.0005
folder: experiment_logs/supervised/
write_tag: linear_probe