diff --git a/lib/scene_parser/rcnn/solver/build.py b/lib/scene_parser/rcnn/solver/build.py
index 865a4ec..e64e0c6 100644
--- a/lib/scene_parser/rcnn/solver/build.py
+++ b/lib/scene_parser/rcnn/solver/build.py
@@ -16,7 +16,7 @@ def make_optimizer(cfg, model):
             weight_decay = cfg.SOLVER.WEIGHT_DECAY_BIAS
         params += [{"params": [value], "lr": lr, "weight_decay": weight_decay}]
 
-    optimizer = torch.optim.SGD(params, lr, momentum=cfg.SOLVER.MOMENTUM)
+    optimizer = torch.optim.SGD(params, cfg.SOLVER.BASE_LR, momentum=cfg.SOLVER.MOMENTUM)
     return optimizer