ColossalAI/colossalai/checkpoint_io
Baizhou Zhang d8ceeac14e
[hotfix] fix typo in hybrid parallel io (#4697)
2023-09-12 17:32:19 +08:00
..
__init__.py [hotfix] fix typo in hybrid parallel io (#4697) 2023-09-12 17:32:19 +08:00
checkpoint_io_base.py Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141) 2023-07-07 16:33:06 +08:00
general_checkpoint_io.py Merge branch 'main' into feature/shardformer 2023-09-04 23:43:13 +08:00
hybrid_parallel_checkpoint_io.py [hotfix] fix typo in hybrid parallel io (#4697) 2023-09-12 17:32:19 +08:00
index_file.py [checkpointio] General Checkpointing of Sharded Optimizers (#3984) 2023-06-15 15:21:26 +08:00
utils.py [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00