From d6a592d7b1f0a28f4b9e1f0bc9984a2a9b639d8a Mon Sep 17 00:00:00 2001
From: Vladislav Sovrasov <sovrasov.vlad@gmail.com>
Date: Thu, 12 Dec 2019 17:23:45 +0300
Subject: [PATCH] Adopt WIDER-SSD300 config to run on 8 GPUs (#1781)
---
configs/wider_face/ssd300_wider_face.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/configs/wider_face/ssd300_wider_face.py b/configs/wider_face/ssd300_wider_face.py
index 6a4184e..9c0ce79 100644
--- 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(
--
GitLab