github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-05 12:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
294a6060d003b7767da4e382f7b8dea7b83d6b6b
ColossalAI/colossalai/zero
History
Jiarui Fang 294a6060d0 [tensor] ZeRO use ColoTensor as the base class. (#828) 
* [refactor] moving InsertPostInitMethodToModuleSubClasses to utils.

* [tensor] ZeRO use ColoTensor as the base class.

* polish
2022-04-22 12:00:48 +08:00
..
init_ctx
[refactor] moving InsertPostInitMethodToModuleSubClasses to utils. (#824)
2022-04-21 16:03:18 +08:00
shard_utils
Revert "[zero] add ZeroTensorShardStrategy (#793)" (#806)
2022-04-19 14:40:02 +08:00
sharded_model
[refactor] moving memtracer to gemini (#801)
2022-04-19 10:13:08 +08:00
sharded_optim
[log] local throughput metrics (#811)
2022-04-20 10:05:39 +08:00
sharded_param
[tensor] ZeRO use ColoTensor as the base class. (#828)
2022-04-22 12:00:48 +08:00
utils
[gemini] collect cpu-gpu moving volume in each iteration (#813)
2022-04-20 11:29:48 +08:00
__init__.py
[log] local throughput metrics (#811)
2022-04-20 10:05:39 +08:00