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-15 23:23:11 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
73d36618a6263b3e0ff3178a4e7dfaaec60655b0
ColossalAI
/
colossalai
/
zero
History
ver217
fb841dd5c5
[zero] optimize grad offload (
#539
)
...
* optimize grad offload * polish code * polish code
2022-03-29 12:48:00 +08:00
..
init_ctx
[zero] polish ZeroInitContext (
#540
)
2022-03-29 09:09:04 +08:00
shard_utils
[zero] refactor model data tracing (
#537
)
2022-03-28 16:38:18 +08:00
sharded_model
[zero] optimize grad offload (
#539
)
2022-03-29 12:48:00 +08:00
sharded_optim
[zero] optimize grad offload (
#539
)
2022-03-29 12:48:00 +08:00
sharded_param
[zero] get memory usage of sharded optim v2. (
#542
)
2022-03-29 09:08:18 +08:00
__init__.py
[refactor] remove old zero code (
#517
)
2022-03-25 14:54:39 +08:00