diff --git a/python/mxnet/module/executor_group.py b/python/mxnet/module/executor_group.py index f98bb4e1e720..01718edc7489 100644 --- 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