ColossalAI/colossalai
jiangmingyan 52a933e175
[checkpoint] support huggingface style sharded checkpoint (#3461)
* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

* [checkpoint] support huggingface style sharded checkpoint

---------

Co-authored-by: luchen <luchen@luchendeMBP.lan>
2023-04-06 16:23:39 +08:00
..
_C
_analyzer
amp
auto_parallel
autochunk
booster [booster] fixed the torch ddp plugin with the new checkpoint api (#3442) 2023-04-06 09:43:51 +08:00
builder
checkpoint_io [checkpoint] support huggingface style sharded checkpoint (#3461) 2023-04-06 16:23:39 +08:00
cli [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
cluster
communication
context
device
engine [format] Run lint on colossalai.engine (#3367) 2023-04-05 23:24:43 +08:00
fx
interface
kernel
logging
nn
pipeline
registry
tensor Fix typo (#3448) 2023-04-06 09:43:31 +08:00
testing [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
trainer
utils [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py