未验证 提交 0bc5b54d 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者: GitHub

fix opt (#1048)

上级 f7ab1c2d
......@@ -119,7 +119,6 @@ class RMSProp(object):
self.epsilon = epsilon
self.weight_decay = weight_decay
self.grad_clip = grad_clip
self.multi_precision = multi_precision
def __call__(self, parameters):
opt = optim.RMSProp(
......@@ -129,6 +128,5 @@ class RMSProp(object):
epsilon=self.epsilon,
weight_decay=self.weight_decay,
grad_clip=self.grad_clip,
multi_precision=self.multi_precision,
parameters=parameters)
return opt
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册