You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_gemini/update
HELSON 56ddc9ca7a
[hotfix] add correct device for fake_param (#2796)
2 years ago
..
test_chunk_mgrv2.py [NFC] update chunk manager API (#2119) 2 years ago
test_chunkv2.py [zero] fix error for BEiT models (#2169) 2 years ago
test_fwd_bwd.py [gemini] update ddp strict mode (#2518) 2 years ago
test_gemini_use_rmt.py [gemini] update ddp strict mode (#2518) 2 years ago
test_get_torch_model.py [gemini] add get static torch model (#2356) 2 years ago
test_grad_clip.py [gemini] update ddp strict mode (#2518) 2 years ago
test_inference.py [gemini] add fake_release_chunk for keep-gathered chunk in the inference mode (#2671) 2 years ago
test_optim.py [gemini] update ddp strict mode (#2518) 2 years ago
test_search.py [gemini] update ddp strict mode (#2518) 2 years ago
test_zeroddp_state_dict.py [gemini] update ddp strict mode (#2518) 2 years ago
test_zerooptim_state_dict.py [hotfix] add correct device for fake_param (#2796) 2 years ago