This website requires JavaScript.
Explore
关于
Help
Register
Sign In
github
/
ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI
Watch
1
Star
0
Fork
You've already forked ColossalAI
0
Code
Issues
Projects
Releases
Wiki
Activity
ec18fc7340
ColossalAI
/
tests
/
test_zero
/
test_gemini
History
Baizhou Zhang
0bb0b481b4
[gemini] fix argument naming during chunk configuration searching
2023-06-25 13:34:15 +08:00
..
test_chunk_mgrv2.py
…
test_chunkv2.py
…
test_fwd_bwd.py
…
test_gemini_use_rmt.py
…
test_get_torch_model.py
…
test_grad_clip.py
…
test_inference.py
…
test_optim.py
…
test_runtime_mem_tracer.py
…
test_search.py
…
test_zeroddp_state_dict.py
…
test_zeroddp_state_dict_shard.py
…
test_zerooptim_state_dict.py
…