未验证 提交 6edbdbfa 编写于 作者: L limingshu 提交者: GitHub

first commit (#36778)

上级 51a33962
...@@ -2068,6 +2068,7 @@ class LarsMomentumOptimizer(Optimizer): ...@@ -2068,6 +2068,7 @@ class LarsMomentumOptimizer(Optimizer):
"lars_coeff": self._lars_coeff, "lars_coeff": self._lars_coeff,
"lars_weight_decay": [_lars_weight_decay], "lars_weight_decay": [_lars_weight_decay],
"multi_precision": find_master, "multi_precision": find_master,
"epsilon": self._epsilon,
"rescale_grad": self._rescale_grad "rescale_grad": self._rescale_grad
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册