From e2b38eabbd5ca9b348de896d85e81c9ec2ee9d62 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Mon, 6 Jul 2026 10:52:23 +0200 Subject: [PATCH] higher batch size --- 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 6200821..8f09488 100644 --- a/configs/wilds_vith16-448_ep300.yaml +++ b/configs/wilds_vith16-448_ep300.yaml @@ -1,5 +1,5 @@ data: - batch_size: 16 + 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: 21 + gradient_accumulation_steps: 2 ema: - 0.996 - 1.0