Commit f7f67961 authored by Yizhou Wang's avatar Yizhou Wang
Browse files

Update config_rodnet_cdc_win16.py

parent 0505a4d8
...@@ -9,11 +9,9 @@ dataset_cfg = dict( ...@@ -9,11 +9,9 @@ dataset_cfg = dict(
'2019_04_09_PMS1000_PL_NORMAL', '2019_04_09_PMS1000_PL_NORMAL',
'2019_04_09_PMS3001_PL_NORMAL', '2019_04_09_PMS3001_PL_NORMAL',
'2019_05_29_MLMS006_CR_BLUR', '2019_05_29_MLMS006_CR_BLUR',
'2019_05_29_PBMS007_PL_BLUR',
'2019_09_29_ONRD001_CS_NORMAL', '2019_09_29_ONRD001_CS_NORMAL',
'2019_09_29_ONRD002_CS_NORMAL', '2019_09_29_ONRD002_CS_NORMAL',
'2019_09_29_ONRD004_HW_NORMAL', '2019_09_29_ONRD004_HW_NORMAL',
'2019_10_13_ONRD048_CS_NIGHT'
], ],
), ),
valid=dict( valid=dict(
...@@ -21,7 +19,8 @@ dataset_cfg = dict( ...@@ -21,7 +19,8 @@ dataset_cfg = dict(
), ),
test=dict( test=dict(
seqs=[ seqs=[
'2019_04_09_BMS1000_PL_NORMAL', '2019_05_29_PBMS007_PL_BLUR',
'2019_10_13_ONRD048_CS_NIGHT'
], ],
), ),
demo=dict( demo=dict(
...@@ -62,13 +61,13 @@ confmap_cfg = dict( ...@@ -62,13 +61,13 @@ confmap_cfg = dict(
) )
train_cfg = dict( train_cfg = dict(
n_epoch=50, n_epoch=10,
batch_size=4, batch_size=4,
lr=0.00001, lr=0.00001,
lr_step=5, # lr will decrease 10 times after lr_step epoches lr_step=5, # lr will decrease 10 times after lr_step epoches
win_size=16, win_size=16,
train_step=1, train_step=1,
train_stride=8, train_stride=16,
log_step=100, log_step=100,
save_step=1000, save_step=1000,
) )
......
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