TST Migrates some of test_nn.py from assertEqualIgnoreTypes to assertEqual (#57642)
Summary:
Related to https://github.com/pytorch/pytorch/issues/38095, https://github.com/pytorch/pytorch/issues/50006
Migrates some of `test_nn.py` from `assertEqualIgnoreTypes` to `assertEqual`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57642
Reviewed By: bdhirsh
Differential Revision: D28317761
Pulled By: mruberry
fbshipit-source-id: 6bea6f669569922b2a391d1523917edde976f014