提交 558f03d6 编写于 作者: Z zhangbo9674

refine code

上级 28061f53
...@@ -64,7 +64,7 @@ class Momentum(object): ...@@ -64,7 +64,7 @@ class Momentum(object):
grad_clip=self.grad_clip, grad_clip=self.grad_clip,
multi_precision=self.multi_precision, multi_precision=self.multi_precision,
parameters=parameters, parameters=parameters,
use_multi_tensor=False) use_multi_tensor=True)
return opt return opt
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册