From 44d7226f8ed7a17c5722e4f3d5e7f517da70aaf4 Mon Sep 17 00:00:00 2001 From: Jin-Hwa Kim Date: Tue, 20 Aug 2019 15:39:37 +0900 Subject: [PATCH] Fix ` 'lr' referenced before assignment` error Does it correct what you've intended? --- lib/scene_parser/rcnn/solver/build.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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