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-05-03 01:19:15 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
7aef75ca42537277ddf1c32a7ebe3b1c11897cf4
ColossalAI
/
tests
/
test_zero_data_parallel
History
ver217
7aef75ca42
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
..
common.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
test_shard_model_v2.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
test_shard_param.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +08:00
test_sharded_optim_with_sync_bn.py
added buffer sync to naive amp model wrapper (
#291
)
2022-03-11 15:50:28 +08:00
test_sharded_optim.py
added unit test for sharded optimizer (
#293
)
2022-03-11 15:50:28 +08:00
test_zero_dev_3_mp4.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
test_zero_dev_3.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +08:00
test_zero_param_mgr.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +08:00
