increase bs to avoid oom

pull/2832/head
gaotingquan 2023-05-31 02:23:40 +00:00 committed by cuicheng01
parent caa6393cd4
commit 07b597f56e
1 changed files with 1 additions and 1 deletions

View File

@ -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: