lock seed for res50_db
This commit is contained in:
parent
e7d981b674
commit
f6cb62b7b8
|
@ -75,10 +75,12 @@ Train:
|
|||
- { 'type': Fliplr, 'args': { 'p': 0.5 } }
|
||||
- { 'type': Affine, 'args': { 'rotate': [-10, 10] } }
|
||||
- { 'type': Resize, 'args': { 'size': [0.5, 3] } }
|
||||
- EastRandomCropData:
|
||||
- Resize:
|
||||
size: [640, 640]
|
||||
max_tries: 50
|
||||
keep_ratio: true
|
||||
# - EastRandomCropData:
|
||||
# size: [640, 640]
|
||||
# max_tries: 50
|
||||
# keep_ratio: true
|
||||
- MakeBorderMap:
|
||||
shrink_ratio: 0.4
|
||||
thresh_min: 0.3
|
||||
|
@ -95,10 +97,10 @@ Train:
|
|||
- KeepKeys:
|
||||
keep_keys: ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask'] # the order of the dataloader list
|
||||
loader:
|
||||
shuffle: True
|
||||
shuffle: False
|
||||
drop_last: False
|
||||
batch_size_per_card: 16
|
||||
num_workers: 8
|
||||
batch_size_per_card: 4
|
||||
num_workers: 2
|
||||
|
||||
Eval:
|
||||
dataset:
|
||||
|
@ -125,4 +127,4 @@ Eval:
|
|||
shuffle: False
|
||||
drop_last: False
|
||||
batch_size_per_card: 1 # must be 1
|
||||
num_workers: 8
|
||||
num_workers: 8
|
||||
|
|
Loading…
Reference in New Issue