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

5 years ago · a6ab43aab4
parent bcdeadcde9
commit a6ab43aab4
1 changed files with 2 additions and 1 deletions
--- 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,