This website requires JavaScript.
Explore
关于
Help
Register
Sign In
github
/
ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI
Watch
1
Star
0
Fork
You've already forked ColossalAI
0
Code
Issues
Projects
Releases
Wiki
Activity
edb67cb378
ColossalAI
/
colossalai
/
zero
/
sharded_model
History
Fazzie-Maqianli
06dccdde44
[NFC] polish colossalai/zero/sharded_model/reduce_scatter.py code style (
#1554
)
2022-09-08 22:11:04 +08:00
..
__init__.py
[refactor] remove old zero code (
#517
)
2022-03-25 14:54:39 +08:00
_utils.py
[gemini] add GeminiMemoryManger (
#832
)
2022-04-24 13:08:48 +08:00
reduce_scatter.py
[NFC] polish colossalai/zero/sharded_model/reduce_scatter.py code style (
#1554
)
2022-09-08 22:11:04 +08:00
sharded_model_v2.py
[hotfix] shared model returns cpu state_dict (
#1328
)
2022-07-15 22:11:37 +08:00
utils.py
[zero] refactor ShardedParamV2 for convenience (
#742
)
2022-04-13 14:54:26 +08:00