Always use NewModuleTest instead of ModuleTest. (#44745)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44745
Much like CriterionTest, NewCriterionTest these are outdated formulations and we should just use the new one.
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D23717808
Pulled By: gchanan
fbshipit-source-id: eb91982eef23452456044381334bfc9a5bbd837e