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-10 15:55:36 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
ce7b2c9ae3cd0cf026175e615b0fac620f535833
ColossalAI
/
colossalai
/
zero
/
sharded_model
History
Jiarui Fang
272ebfb57d
[bug] shard param during initializing the ShardedModelV2 (
#381
)
2022-03-11 15:50:28 +08:00
..
__init__.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +08:00
_zero3_utils.py
[zero] Update sharded model v2 using sharded param v2 (
#323
)
2022-03-11 15:50:28 +08:00
param_manager.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +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_grad.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
sharded_model_v2.py
[bug] shard param during initializing the ShardedModelV2 (
#381
)
2022-03-11 15:50:28 +08:00
sharded_model.py
[zero] Update sharded model v2 using sharded param v2 (
#323
)
2022-03-11 15:50:28 +08:00
utils.py
[test] polish zero related unitest (
#351
)
2022-03-11 15:50:28 +08:00
