Fix bind_ith_exec calling simple_bind

reminisce · May 30, 2017 · 1aae9dc · 1aae9dc
1 parent cc7aef9
commit 1aae9dc
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
@@ -578,9 +578,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