Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
2ad0ca44
Commit
2ad0ca44
authored
Dec 16, 2020
by
WenmuZhou
Browse files
update db config
parent
d3b609ee
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
5 deletions
+5
-5
configs/det/det_mv3_db.yml
configs/det/det_mv3_db.yml
+4
-4
configs/det/det_r50_vd_db.yml
configs/det/det_r50_vd_db.yml
+1
-1
No files found.
configs/det/det_mv3_db.yml
View file @
2ad0ca44
...
@@ -2,7 +2,7 @@ Global:
...
@@ -2,7 +2,7 @@ Global:
use_gpu
:
true
use_gpu
:
true
epoch_num
:
1200
epoch_num
:
1200
log_smooth_window
:
20
log_smooth_window
:
20
print_batch_step
:
2
print_batch_step
:
10
save_model_dir
:
./output/db_mv3/
save_model_dir
:
./output/db_mv3/
save_epoch_step
:
1200
save_epoch_step
:
1200
# evaluation is run every 5000 iterations after the 4000th iteration
# evaluation is run every 5000 iterations after the 4000th iteration
...
@@ -39,7 +39,7 @@ Loss:
...
@@ -39,7 +39,7 @@ Loss:
alpha
:
5
alpha
:
5
beta
:
10
beta
:
10
ohem_ratio
:
3
ohem_ratio
:
3
Optimizer
:
Optimizer
:
name
:
Adam
name
:
Adam
beta1
:
0.9
beta1
:
0.9
...
@@ -100,7 +100,7 @@ Train:
...
@@ -100,7 +100,7 @@ Train:
loader
:
loader
:
shuffle
:
True
shuffle
:
True
drop_last
:
False
drop_last
:
False
batch_size_per_card
:
4
batch_size_per_card
:
16
num_workers
:
8
num_workers
:
8
Eval
:
Eval
:
...
@@ -128,4 +128,4 @@ Eval:
...
@@ -128,4 +128,4 @@ Eval:
shuffle
:
False
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
:
8
\ No newline at end of file
\ No newline at end of file
configs/det/det_r50_vd_db.yml
View file @
2ad0ca44
...
@@ -6,7 +6,7 @@ Global:
...
@@ -6,7 +6,7 @@ Global:
save_model_dir
:
./output/det_r50_vd/
save_model_dir
:
./output/det_r50_vd/
save_epoch_step
:
1200
save_epoch_step
:
1200
# evaluation is run every 5000 iterations after the 4000th iteration
# evaluation is run every 5000 iterations after the 4000th iteration
eval_batch_step
:
[
5
000
,
4
000
]
eval_batch_step
:
[
4
000
,
5
000
]
# if pretrained_model is saved in static mode, load_static_weights must set to True
# if pretrained_model is saved in static mode, load_static_weights must set to True
load_static_weights
:
True
load_static_weights
:
True
cal_metric_during_train
:
False
cal_metric_during_train
:
False
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment