change to int to compatible with py3

7 years ago · f42247e55f
parent 0707abb51b
commit f42247e55f
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@ -878,7 +878,7 @@ class DistributeTranspiler(object):
        origin_param_var = self.origin_program.global_block().vars[
            self.table_name]

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