Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Correct init idx2name on mult-CPU/GPU training#2337

Merged
piiswrong merged 1 commit intoapache:masterfrom
Answeror:bug/idx2name
Jun 4, 2016
Merged

Correct init idx2name on mult-CPU/GPU training#2337
piiswrong merged 1 commit intoapache:masterfrom
Answeror:bug/idx2name

Conversation

@Answeror
Copy link
Contributor

@Answeror Answeror commented Jun 4, 2016

In Module.init_optimizer, idx2name has duplicated items when multiple contexts are provided.
But _update_params_on_kvstore will be used in this case, and the duplication of items will cause wrong indexing in Optimizer._get_lr and Optimizer._get_wd.

In Module.init_optimizer, idx2name has duplicated items when multiple
contexts are provided.
But _update_params_on_kvstore will be used in this case, and the
duplication of items will cause wrong indexing in Optimizer._get_lr and
Optimizer._get_wd.
@piiswrong
Copy link
Contributor

LGTM. Thanks

@piiswrong piiswrong merged commit b1eef81 into apache:master Jun 4, 2016
Answeror added a commit to Answeror/mxnet that referenced this pull request Jun 7, 2016
This patch is related to apache#2327 and apache#2337, which fix a bug of idx2name in multi-GPU environment for Module.
Note, all previous codes which involved lr_mult and wd_mult in multi-GPU environment may suffer from this bug,
for example, finetuning with lr_mult of lower layers fixed to zero on multi-GPU will give incorrect result.
tqchen pushed a commit that referenced this pull request Jun 8, 2016
* Correct init idx2name on multi-CPU/GPU training v2

This patch is related to #2327 and #2337, which fix a bug of idx2name in multi-GPU environment for Module.
Note, all previous codes which involved lr_mult and wd_mult in multi-GPU environment may suffer from this bug,
for example, finetuning with lr_mult of lower layers fixed to zero on multi-GPU will give incorrect result.

* Refine ResNet example
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants