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-29 21:49:54 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
ab8c6b4a0ec35613618ce1e96d5d9ad7a7320a1b
ColossalAI
/
colossalai
/
zero
/
sharded_model
History
ver217
ab8c6b4a0e
[zero] refactor memstats collector (
#706
)
...
* refactor memstats collector * fix disposable * polish code
2022-04-11 10:46:08 +08:00
..
__init__.py
[refactor] remove old zero code (
#517
)
2022-03-25 14:54:39 +08:00
_utils.py
[zero] add stateful tensor (
#549
)
2022-03-30 13:51:37 +08:00
reduce_scatter.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
sharded_model_v2.py
[zero] refactor memstats collector (
#706
)
2022-04-11 10:46:08 +08:00
utils.py
[polish] rename col_attr -> colo_attr (
#558
)
2022-03-31 12:25:45 +08:00