ColossalAI/tests/test_gemini/update
Jiarui Fang 05bb28aacf
[Gemini] mapping of preop timestep and param (#2124)
2022-12-13 12:50:24 +08:00
..
test_chunk_mgrv2.py [NFC] update chunk manager API (#2119) 2022-12-12 16:57:22 +08:00
test_chunkv2.py [NFC] polish comments for Chunk class (#2116) 2022-12-12 15:39:31 +08:00
test_fwd_bwd.py [Gemini] remove eval in gemini unittests! (#2092) 2022-12-07 11:58:37 +08:00
test_gemini_use_rmt.py [Gemini] mapping of preop timestep and param (#2124) 2022-12-13 12:50:24 +08:00
test_grad_clip.py [zero] add L2 gradient clipping for ZeRO (#2112) 2022-12-09 18:09:17 +08:00
test_optim.py [Gemini] chunk init using runtime visited param order (#2115) 2022-12-12 18:06:16 +08:00
test_search.py
test_zeroddp_state_dict.py [Gemini] more tests for Gemini (#2038) 2022-11-29 17:13:10 +08:00
test_zerooptim_state_dict.py