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

sgd

Browse Source
This commit is contained in:
YannAhlgrim
2026-05-22 11:37:19 +02:00
parent 142b6beec7
commit 5f2d7d224d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/supervised_wilds_vith14_ep300-lp.yaml
+2 -2
View File
@@ -24,7 +24,7 @@ mask:
patch_size: 14 patch_size: 14
optimization: optimization:
optimizer: lars # 'adamw', 'sgd', or 'lars' optimizer: sgd # 'adamw', 'sgd', or 'lars'
freeze_weights: true # true for linear probing, false for full fine-tuning freeze_weights: true # true for linear probing, false for full fine-tuning
epochs: 300 epochs: 300
lr: 0.01 lr: 0.01
@@ -55,5 +55,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
folder: experiment_logs/supervised-lars/ folder: experiment_logs/supervised-sgd/
write_tag: linear_probe write_tag: linear_probe