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

d6a592d7 · Vladislav Sovrasov · Kai Chen · 69d62cfd · d6a592d7
Commit d6a592d7 authored Dec 12, 2019 by Vladislav Sovrasov Committed by Kai Chen Dec 12, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

configs/wider_face/ssd300_wider_face.py configs/wider_face/ssd300_wider_face.py +1 -1

No files found.
--- a/configs/wider_face/ssd300_wider_face.py
+++ b/configs/wider_face/ssd300_wider_face.py
@@ -107,7 +107,7 @@ data = dict(
        img_prefix=data_root + 'WIDER_val/',
        pipeline=test_pipeline))
 # 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()
 # learning policy
 lr_config = dict(