Safeguard the learning rate map hack

b4614894 · Yang Zhang · 4e9c403e · b4614894
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

model.py model.py +1 -1

未找到文件。
--- a/model.py
+++ b/model.py
@@ -280,7 +280,7 @@ class StaticGraphAdapter(object):

    def _make_program(self, inputs):
        prog = self._orig_prog.clone(for_test=self.mode != 'train')
-        if self.mode == 'train':
+        if self.mode == 'train' 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