[BE] Fix FSDP flaky checkpoint test

Commit

2 years ago

[BE] Fix FSDP flaky checkpoint test Pull Request resolved: https://github.com/pytorch/pytorch/pull/78856 Approved by: https://github.com/awgu