ColossalAI/tests/test_gemini/update
HELSON 707b11d4a0
[gemini] update ddp strict mode (#2518)
* [zero] add strict ddp mode for chunk init

* [gemini] update gpt example
2023-01-28 14:35:25 +08:00
..
test_chunk_mgrv2.py [NFC] update chunk manager API (#2119) 2022-12-12 16:57:22 +08:00
test_chunkv2.py [zero] fix error for BEiT models (#2169) 2022-12-26 15:03:54 +08:00
test_fwd_bwd.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_gemini_use_rmt.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_get_torch_model.py [gemini] add get static torch model (#2356) 2023-01-06 13:41:19 +08:00
test_grad_clip.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_inference.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_optim.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_search.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_zeroddp_state_dict.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00
test_zerooptim_state_dict.py [gemini] update ddp strict mode (#2518) 2023-01-28 14:35:25 +08:00