提交 caa6393c 编写于 作者: G gaotingquan 提交者: cuicheng01

set drop_last to False in train data

上级 cf5d629a
...@@ -98,7 +98,7 @@ DataLoader: ...@@ -98,7 +98,7 @@ DataLoader:
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 128 batch_size: 128
drop_last: True drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 4 num_workers: 4
......
...@@ -98,7 +98,7 @@ DataLoader: ...@@ -98,7 +98,7 @@ DataLoader:
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 128 batch_size: 128
drop_last: True drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 4 num_workers: 4
......
...@@ -98,7 +98,7 @@ DataLoader: ...@@ -98,7 +98,7 @@ DataLoader:
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 128 batch_size: 128
drop_last: True drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 4 num_workers: 4
......
...@@ -98,7 +98,7 @@ DataLoader: ...@@ -98,7 +98,7 @@ DataLoader:
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 128 batch_size: 128
drop_last: True drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 4 num_workers: 4
......
...@@ -98,7 +98,7 @@ DataLoader: ...@@ -98,7 +98,7 @@ DataLoader:
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 128 batch_size: 128
drop_last: True drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 4 num_workers: 4
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册