Fix bind_ith_exec calling simple_bind

apache · May 23, 2017 · e8b77b5 · e8b77b5
1 parent 23a2d06
commit e8b77b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/module/executor_group.py b/python/mxnet/module/executor_group.py
@@ -573,9 +573,9 @@ def _bind_ith_exec(self, i, data_shapes, label_shapes, shared_group):
             input_types.update({x.name: x.dtype for x in label_shapes})
 
         executor = self.symbol.simple_bind(ctx=context, grad_req=self.grad_req,
-                                           type_dict=input_types, param_names=self.param_names,
+                                           type_dict=input_types, shared_arg_names=self.param_names,
                                            shared_exec=shared_exec,
-                                           shared_data_arrays=shared_data_arrays, **input_shapes)
+                                           shared_buffer=shared_data_arrays, **input_shapes)
         self._total_exec_bytes += int(executor.debug_str().split('\n')[-3].split()[1])
         return executor