Fix ZeRO3 save_checkpoint (#857)

Commit

4 years ago

Fix ZeRO3 save_checkpoint (#857) Co-authored-by: Jeff Rasley <jerasley@microsoft.com>