From 2551d8dd40383fb4cfc86cc9a1fc034436d29bc0 Mon Sep 17 00:00:00 2001 From: Yann Ahlgrim Date: Tue, 9 Jun 2026 00:26:55 +0200 Subject: [PATCH] load checkpoint false and increase batch size (2048) --- 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 9cc9e58..eb78488 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: 512 color_jitter_strength: 0.0 crop_scale: - 1.0 @@ -32,7 +32,7 @@ mask: - 0.2 meta: copy_data: false - load_checkpoint: true + load_checkpoint: false model_name: vit_huge pred_depth: 12 pred_emb_dim: 384