Update distribute_transpiler.py

7 years ago · 7dbf5f1244
parent c0a95bb2d8
commit 7dbf5f1244
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
@ -712,7 +712,7 @@ class DistributeTranspiler(object):
        slice_vars_and_attrs = []
        block_suffix = "block"
        for param in self.param_grad_ep_mapping[endpoint]["params"]:
-            orig_var_name, block_name, _ = self._get_varname_parts(param)
+            orig_var_name, block_name, _ = self._get_varname_parts(param.name)
            if not block_name:
                continue