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

reduce batch size

Browse Source
This commit is contained in:
YannAhlgrim
2026-05-18 11:04:47 +02:00
parent bd8debf88f
commit 8e5815923a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/wilds_vitb16-448_ep300.yaml
+2 -2
View File
@@ -1,5 +1,5 @@
data: data:
batch_size: 128 batch_size: 16
color_jitter_strength: 0.0 color_jitter_strength: 0.0
crop_scale: crop_scale:
- 1.0 - 1.0
@@ -43,7 +43,7 @@ optimization:
ema: ema:
- 0.996 - 0.996
- 1.0 - 1.0
epochs: 300 epochs: 600
final_lr: 1.0e-06 final_lr: 1.0e-06
final_weight_decay: 0.4 final_weight_decay: 0.4
ipe_scale: 1.0 ipe_scale: 1.0