Skip to content
Snippets Groups Projects
Commit d6a592d7 authored by Vladislav Sovrasov's avatar Vladislav Sovrasov Committed by Kai Chen
Browse files

Adopt WIDER-SSD300 config to run on 8 GPUs (#1781)

parent 69d62cfd
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
configs/wider_face/ssd300_wider_face.py
+ 1
− 1
View file @ d6a592d7
...@@ -107,7 +107,7 @@ data = dict( ...@@ -107,7 +107,7 @@ data = dict(
img_prefix=data_root + 'WIDER_val/', img_prefix=data_root + 'WIDER_val/',
pipeline=test_pipeline)) pipeline=test_pipeline))
# optimizer # optimizer
optimizer = dict(type='SGD', lr=1e-3, momentum=0.9, weight_decay=5e-4) optimizer = dict(type='SGD', lr=0.012, momentum=0.9, weight_decay=5e-4)
optimizer_config = dict() optimizer_config = dict()
# learning policy # learning policy
lr_config = dict( lr_config = dict(
......
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