Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 01:03:35 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c97a3523db0db7048cbd14942825183f0ea66fe3
ColossalAI/colossalai/checkpoint_io
History
Baizhou Zhang 64a08b2dc3 [checkpointio] support unsharded checkpointIO for hybrid parallel (#4774)
* support unsharded saving/loading for model

* support optimizer unsharded saving

* update doc

* support unsharded loading for optimizer

* small fix
2023-09-26 10:58:03 +08:00
..
__init__.py
[misc] update pre-commit and run all files (#4752)
2023-09-19 14:20:26 +08:00
checkpoint_io_base.py
[shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758)
2023-09-20 18:29:37 +08:00
general_checkpoint_io.py
[shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758)
2023-09-20 18:29:37 +08:00
hybrid_parallel_checkpoint_io.py
[checkpointio] support unsharded checkpointIO for hybrid parallel (#4774)
2023-09-26 10:58:03 +08:00
index_file.py
[misc] update pre-commit and run all files (#4752)
2023-09-19 14:20:26 +08:00
utils.py
[shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758)
2023-09-20 18:29:37 +08:00
Powered by Gitea Version: 1.25.2 Page: 424ms Template: 55ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API