ColossalAI/tests/test_gemini/update
HELSON 48d33b1b17
[gemini] add get static torch model (#2356)
2023-01-06 13:41:19 +08:00
..
test_chunk_mgrv2.py
test_chunkv2.py
test_fwd_bwd.py
test_gemini_use_rmt.py
test_get_torch_model.py [gemini] add get static torch model (#2356) 2023-01-06 13:41:19 +08:00
test_grad_clip.py
test_optim.py [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_search.py
test_zeroddp_state_dict.py
test_zerooptim_state_dict.py