Back out "Revert D30745961: [DDP] Remove self.modules_params" (#64777)
authorRohan Varma <rvarm1@fb.com>
Fri, 17 Sep 2021 19:23:09 +0000 (12:23 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 17 Sep 2021 19:28:34 +0000 (12:28 -0700)
commit70f286c1e2ec6fda8b1765d77a6dafeb7c8b0add
tree2f6af9a1a081a683ced07d97677f41a87a83ffcf
parent61dfcbf4bcc14dbd08b316281d29bcb5c013c398
Back out "Revert D30745961: [DDP] Remove self.modules_params" (#64777)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64777

Original commit changeset: 59f7cc50d369
ghstack-source-id: 138326909

Test Plan: ci

Reviewed By: H-Huang

Differential Revision: D30849442

fbshipit-source-id: bb87ba83935374d8a3ebbc29365df1417dd4f26f
torch/nn/parallel/distributed.py