reduce batch size because of GPU overload

This commit is contained in:
YannAhlgrim
2026-05-16 11:20:41 +02:00
parent 7101f6c8e1
commit bad00f306e
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 128 batch_size: 32
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true