You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/checkpoint_io
wukong1992 6b305a99d6
[booster] torch fsdp fix ckpt (#3788)
2 years ago
..
__init__.py
checkpoint_io_base.py [plugin] torch ddp plugin supports sharded model checkpoint (#3775) 2 years ago
general_checkpoint_io.py [booster] torch fsdp fix ckpt (#3788) 2 years ago
index_file.py
utils.py [plugin] torch ddp plugin supports sharded model checkpoint (#3775) 2 years ago