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

augment for 6 GPUs

Browse Source
This commit is contained in:
YannAhlgrim
2026-06-15 13:40:10 +02:00
parent ccdddc47e5
commit 8b714927ab
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/wilds_vith16-448_ep300.yaml
+2 -2
View File
@@ -32,7 +32,7 @@ mask:
- 0.2 - 0.2
meta: meta:
copy_data: false copy_data: false
load_checkpoint: false load_checkpoint: true
model_name: vit_huge model_name: vit_huge
pred_depth: 12 pred_depth: 12
pred_emb_dim: 384 pred_emb_dim: 384
@@ -40,7 +40,7 @@ meta:
use_bfloat16: true use_bfloat16: true
use_gradient_checkpointing: true use_gradient_checkpointing: true
optimization: optimization:
gradient_accumulation_steps: 32 gradient_accumulation_steps: 21
ema: ema:
- 0.996 - 0.996
- 1.0 - 1.0