FSDP.scatter_full_optim_state_dict get blocked

When I try to load the state dict, the process will take at least 30 minutes until the NCCL timeout error occurs.

This is my codes: