Should assign norm_type instead of scale_grad_by_freq

[brad-mengchi]

What does this PR do?

Fixes #1156 (issue).

Before submitting

• [x] Did you have fun?
  • Make sure you had fun coding 🙃
• [x] Did you read the contributor guideline?
• [x] Was this discussed/approved via a Github issue? (no need for typos, doc improvements)
  • [ ] N/A
• [x] Did you make sure to update the docs?
  • [x] N/A
• [ ] Did you write any new necessary tests?
  • [ ] N/A
• [ ] Did you update the changelog? (if needed)
  • [x] N/A

PR review

Anyone in the community is free to review the PR once the tests have passed. If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

[brad-mengchi]

Already fixed by https://github.com/facebookresearch/fairscale/pull/1157. so close this.