github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 17:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
a463980aab30116a4b0091478789d02cacf3d30d
ColossalAI/tests/test_zero_data_parallel
History
Frank Lee a463980aab added unit test for sharded optimizer (#293) 
* added unit test for sharded optimizer

* refactor for elegance
2022-03-02 17:15:54 +08:00
..
common.py
Feature/zero (#279)
2022-03-01 18:17:01 +08:00
test_shard_model_v2.py
Feature/zero (#279)
2022-03-01 18:17:01 +08:00
test_shard_param.py
Feature/zero (#279)
2022-03-01 18:17:01 +08:00
test_sharded_optim_with_sync_bn.py
added buffer sync to naive amp model wrapper (#291)
2022-03-02 16:47:17 +08:00
test_sharded_optim.py
added unit test for sharded optimizer (#293)
2022-03-02 17:15:54 +08:00
test_zero_dev_3_mp4.py
Feature/zero (#279)
2022-03-01 18:17:01 +08:00
test_zero_dev_3.py
Feature/zero (#279)
2022-03-01 18:17:01 +08:00
test_zero_param_mgr.py
Feature/zero (#279)
2022-03-01 18:17:01 +08:00