fix fsdp checkpointing issues (#24926)

Commit

2 years ago

fix fsdp checkpointing issues (#24926) * fix fsdp load * Update trainer.py * remove saving duplicate state_dict