increase bs to avoid oom
parent
caa6393cd4
commit
07b597f56e
|
@ -107,7 +107,7 @@ DataLoader:
|
||||||
order: ''
|
order: ''
|
||||||
sampler:
|
sampler:
|
||||||
name: DistributedBatchSampler
|
name: DistributedBatchSampler
|
||||||
batch_size: 128
|
batch_size: 64
|
||||||
drop_last: False
|
drop_last: False
|
||||||
shuffle: False
|
shuffle: False
|
||||||
loader:
|
loader:
|
||||||
|
|
Loading…
Reference in New Issue