Deep-Feature-Flow icon indicating copy to clipboard operation
Deep-Feature-Flow copied to clipboard

Lack a parameter in _update_params_on_kvstore() function

Open Fuckmi opened this issue 6 years ago • 1 comments

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)

Fuckmi avatar Aug 31 '17 01:08 Fuckmi

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)

yuanyao366 avatar Dec 05 '17 17:12 yuanyao366