未验证 提交 a6ab43aa 编写于 作者: S ShenLiang 提交者: GitHub

add type for lsgd, test=develop (#24702)

上级 bcdeadcd
......@@ -314,7 +314,8 @@ class LocalSGD(Collective):
name=self.snapshot_name(param.name),
shape=param.shape,
persistable=True,
stop_gradient=True)
stop_gradient=True,
dtype=param.dtype)
block._insert_op(
idx + 1,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册