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