Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-21 06:22:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b5f43acee3e432a174ecd48c5e128ca28de6da58
ColossalAI/tests/test_zero_data_parallel
History
Jiarui Fang 6b6002962a [zero] zero init context collect numel of model (#375)
2022-03-11 15:50:28 +08:00
..
common.py
[test] polish zero related unitest (#351)
2022-03-11 15:50:28 +08:00
test_init_context.py
[zero] zero init context collect numel of model (#375)
2022-03-11 15:50:28 +08:00
test_shard_model_v2.py
[zero] able to place params on cpu after zero init context (#365)
2022-03-11 15:50:28 +08:00
test_shard_param.py
[zero] Update sharded model v2 using sharded param v2 (#323)
2022-03-11 15:50:28 +08:00
test_sharded_optim_v2_with_cpu_adam.py
[test] polish zero related unitest (#351)
2022-03-11 15:50:28 +08:00
test_sharded_optim_v2.py
[zero] update sharded optim v2 (#334)
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_state_dict.py
[zero] Update sharded model v2 using sharded param v2 (#323)
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
using pytest parametrize
2022-03-11 15:50:28 +08:00
Powered by Gitea Version: 1.25.2 Page: 307ms Template: 13ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API