reduce batch size because of GPU overload
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user