Merge pull request PaddlePaddle#26 from willthefrog/fix_lr_scaling

Fix learning rate scaling bug
qingqing01 · Apr 3, 2020 · 44ba39b · 44ba39b
2 parents d8d5176 + 810ece8
commit 44ba39b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/model.py b/model.py
@@ -410,7 +410,8 @@ def _make_program(self, mode):
                 and self.model._optimizer._learning_rate_map:
             # HACK workaround learning rate map issue
             lr_var = self.model._optimizer._learning_rate_map[self._orig_prog]
-            self.model._optimizer._learning_rate_map[prog] = lr_var
+            new_lr_var = prog.global_block().vars[lr_var.name]
+            self.model._optimizer._learning_rate_map[prog] = new_lr_var
 
         losses = []
         metrics = []