Lack a parameter in _update_params_on_kvstore() function

[jaxfield]

When I run dff_rfcn_end2end_train_test.py as instruction, it comes the error that in _update_params_on_kvstore() function lack a parameter.

Traceback (most recent call last): File "experiments/dff_rfcn/dff_rfcn_end2end_train_test.py", line 19, in train_end2end.main() File "experiments/dff_rfcn/../../dff_rfcn/train_end2end.py", line 171, in main config.TRAIN.begin_epoch, config.TRAIN.end_epoch, config.TRAIN.lr, config.TRAIN.lr_step) File "experiments/dff_rfcn/../../dff_rfcn/train_end2end.py", line 164, in train_net arg_params=arg_params, aux_params=aux_params, begin_epoch=begin_epoch, num_epoch=end_epoch) File "experiments/dff_rfcn/../../dff_rfcn/core/module.py", line 974, in fit self.update() File "experiments/dff_rfcn/../../dff_rfcn/core/module.py", line 1056, in update self._curr_module.update() File "experiments/dff_rfcn/../../dff_rfcn/core/module.py", line 572, in update self._kvstore TypeError: _update_params_on_kvstore() takes exactly 4 arguments (3 given)

[yuanyao366]

This problem could be solved after adding one input parameter for the _update_params_on_kvstore function in rfcn/core/module.py

_update_params_on_kvstore(self._exec_group.param_arrays, self._exec_group.grad_arrays, self._kvstore, self._param_names)