fix: fix batch size to 64 * 4
parent
10559616ba
commit
720c451bc4
|
@ -78,7 +78,7 @@ DataLoader:
|
||||||
|
|
||||||
sampler:
|
sampler:
|
||||||
name: DistributedBatchSampler
|
name: DistributedBatchSampler
|
||||||
batch_size: 32
|
batch_size: 64
|
||||||
drop_last: False
|
drop_last: False
|
||||||
shuffle: True
|
shuffle: True
|
||||||
loader:
|
loader:
|
||||||
|
|
Loading…
Reference in New Issue