From 35d92f12197513501f836ac64f1f5c504d53b697 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Fri, 17 Jul 2026 10:04:19 +0200 Subject: [PATCH] ram overflow --- configs/wilds_vith16-448_ep300.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/wilds_vith16-448_ep300.yaml b/configs/wilds_vith16-448_ep300.yaml index 06ac890..cb831f9 100644 --- a/configs/wilds_vith16-448_ep300.yaml +++ b/configs/wilds_vith16-448_ep300.yaml @@ -1,5 +1,5 @@ data: - batch_size: 256 + batch_size: 128 color_jitter_strength: 0.0 crop_scale: - 1.0 @@ -40,7 +40,7 @@ meta: use_bfloat16: true use_gradient_checkpointing: true optimization: - gradient_accumulation_steps: 1 + gradient_accumulation_steps: 2 ema: - 0.996 - 1.0