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-07-13 03:38:19 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
0db43fa995a8070e69cfce3038dcdcb9a13ba90a
ColossalAI
/
colossalai
/
zero
History
Jiarui Fang
00670c870e
[zero] bucketized tensor cpu gpu copy (
#368
)
2022-03-11 15:50:28 +08:00
..
init_ctx
[zero] able to place params on cpu after zero init context (
#365
)
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] bucketized tensor cpu gpu copy (
#368
)
2022-03-11 15:50:28 +08:00
sharded_param
[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
