This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2025-10-21 23:02:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
6b6002962a960ba18d23deaa58bc087159c31970
ColossalAI
/
colossalai
/
zero
History
Jiarui Fang
6b6002962a
[zero] zero init context collect numel of model (
#375
)
2022-03-11 15:50:28 +08:00
..
init_ctx
[zero] zero init context collect numel of model (
#375
)
2022-03-11 15:50:28 +08:00
shard_utils
[zero] able to place params on cpu after zero init context (
#365
)
2022-03-11 15:50:28 +08:00
sharded_model
fix grad shape
2022-03-11 15:50:28 +08:00
sharded_optim
Revert "[zero] bucketized tensor cpu gpu copy (
#368
)"
2022-03-11 15:50:28 +08:00
sharded_param
Revert "[zero] bucketized tensor cpu gpu copy (
#368
)"
2022-03-11 15:50:28 +08:00
__init__.py
added buffer sync to naive amp model wrapper (
#291
)
2022-03-11 15:50:28 +08:00
