Skip to content

GitLab

Commit 33c66f8e authored by WenmuZhou's avatar WenmuZhou
Browse files

add shuffle param

parent 2aa92e6a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
configs/det/det_mv3_db.yml
View file @ 33c66f8e
...@@ -100,6 +100,7 @@ Train: ...@@ -100,6 +100,7 @@ Train:
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask'] # the order of the dataloader list keep_keys: ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask'] # the order of the dataloader list
loader: loader:
shuffle: True
drop_last: False drop_last: False
batch_size_per_card: 4 batch_size_per_card: 4
num_workers: 8 num_workers: 8
...@@ -126,6 +127,7 @@ Eval: ...@@ -126,6 +127,7 @@ Eval:
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'shape', 'polys', 'ignore_tags'] keep_keys: ['image', 'shape', 'polys', 'ignore_tags']
loader: loader:
shuffle: False
drop_last: False drop_last: False
batch_size_per_card: 1 # must be 1 batch_size_per_card: 1 # must be 1
num_workers: 2 num_workers: 2
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment