ColossalAI/colossalai/zero
Jianghai 376533a564
[shardformer] zero1+pp and the corresponding tests (#4517)
* pause

* finish pp+zero1

* Update test_shard_vit.py
2023-08-28 10:51:16 +08:00
..
gemini [shardformer] support sharded checkpoint IO for models of HybridParallelPlugin (#4506) 2023-08-25 22:04:57 +08:00
legacy [nfc] fix typo colossalai/zero (#3923) 2023-06-08 00:01:29 +08:00
low_level [shardformer] zero1+pp and the corresponding tests (#4517) 2023-08-28 10:51:16 +08:00
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
wrapper.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00