error bug on saving distributed optim state when using data parallel
the indexing typo causes a wrong result when saving distributed optimizer when enabling data parallelism.

cc @pacman100 <|||||>_The documentation is not available anymore as the PR was closed or merged._