fix a bug

7 years ago · 77fffc60c6
parent 5383f3c445
commit 77fffc60c6
1 changed files with 5 additions and 2 deletions
--- a/python/paddle/v2/fluid/backward.py
+++ b/python/paddle/v2/fluid/backward.py
@ -13,8 +13,11 @@ def _rename_arg_(op_desc_list, old_name, new_name, begin_idx=None,
    if end_idx is None:
        end_idx = len(op_desc_list)
    for i in range(begin_idx, end_idx):
-        op_desc_list[i].rename_input(old_name, new_name)
-        op_desc_list[i].rename_output(old_name, new_name)
+        op_desc = op_desc_list[i]
+        if isinstance(op_desc, tuple):
+            op_desc = op_desc[0]
+        op_desc.rename_input(old_name, new_name)
+        op_desc.rename_output(old_name, new_name)


 def _create_op_desc_(op_type, inputs, outputs, attrs):