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

a6ab43aa · ShenLiang · GitHub · bcdeadcd · a6ab43aa
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/transpiler/collective.py python/paddle/fluid/transpiler/collective.py +2 -1

未找到文件。
--- a/python/paddle/fluid/transpiler/collective.py
+++ b/python/paddle/fluid/transpiler/collective.py
@@ -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,