[FSDP] Enhance test for checkpoint
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75236
Make sure that the param names are such as we can load from checkpoint
into a local model.
Differential Revision: [D35377774](https://our.internmc.facebook.com/intern/diff/D35377774/)
Approved by: https://github.com/awgu