[FSDP] Improve `test_fsdp_hybrid_shard_basic_setup` (#106072)
Differential Revision: [D47852460](https://our.internmc.facebook.com/intern/diff/D47852460)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106072
Approved by: https://github.com/rohan-varma
