Merge pull request #13186 from seiriosPlus/selected_rows_size_fix

bug fix
7 years ago · 6bc16ff33b
parent 3b6090e80b ff618a2880
commit 6bc16ff33b
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@ -1096,7 +1096,8 @@ class DistributeTranspiler(object):
            self.table_name]

        zero_dim = int(
-            math.ceil(origin_param_var.shape[0] / len(self.pserver_endpoints)))
+            math.ceil(origin_param_var.shape[0] / float(
+                len(self.pserver_endpoints))))
        table_shape = list(origin_param_var.shape)
        table_shape[0] = zero_dim