add shuffle param

This commit is contained in:
WenmuZhou 2020-11-06 19:02:04 +08:00
parent 2aa92e6a64
commit 33c66f8ed3
1 changed files with 2 additions and 0 deletions

View File

@ -100,6 +100,7 @@ Train:
- KeepKeys:
keep_keys: ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask'] # the order of the dataloader list
loader:
shuffle: True
drop_last: False
batch_size_per_card: 4
num_workers: 8
@ -126,6 +127,7 @@ Eval:
- KeepKeys:
keep_keys: ['image', 'shape', 'polys', 'ignore_tags']
loader:
shuffle: False
drop_last: False
batch_size_per_card: 1 # must be 1
num_workers: 2