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

refine code

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