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
2026-01-25 23:04:48 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
b5f43acee3e432a174ecd48c5e128ca28de6da58
ColossalAI
/
colossalai
/
zero
History
Jiarui Fang
b5f43acee3
[zero] find miss code (
#378
)
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
[zero] find miss code (
#378
)
2022-03-11 15:50:28 +08:00
sharded_param
[zero] find miss code (
#378
)
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