From bad00f306ea069b015a95a82a6a0505bb3406088 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Sat, 16 May 2026 11:20:41 +0200 Subject: [PATCH] reduce batch size because of GPU overload --- configs/supervised_wilds_vith14_ep300-ft.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/supervised_wilds_vith14_ep300-ft.yaml b/configs/supervised_wilds_vith14_ep300-ft.yaml index e4d145f..8683f43 100644 --- a/configs/supervised_wilds_vith14_ep300-ft.yaml +++ b/configs/supervised_wilds_vith14_ep300-ft.yaml @@ -8,7 +8,7 @@ meta: num_classes: 182 data: - batch_size: 128 + batch_size: 32 root_path: ./wilds_data num_workers: 10 pin_mem: true