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.

Jiarui Fang c89c66a858 [Gemini] update API of the chunkmemstatscollector. (#2129)		2 years ago
..
test_chunk_mgrv2.py	[NFC] update chunk manager API (#2119)	2 years ago
test_chunkv2.py	[NFC] polish comments for Chunk class (#2116)	2 years ago
test_fwd_bwd.py	[Gemini] remove eval in gemini unittests! (#2092)	2 years ago
test_gemini_use_rmt.py	[Gemini] update API of the chunkmemstatscollector. (#2129)	2 years ago
test_grad_clip.py	[zero] add L2 gradient clipping for ZeRO (#2112)	2 years ago
test_optim.py	[Gemini] chunk init using runtime visited param order (#2115)	2 years ago
test_search.py	[zero] add chunk init function for users (#1729)	2 years ago
test_zeroddp_state_dict.py	[Gemini] more tests for Gemini (#2038)	2 years ago
test_zerooptim_state_dict.py	[Gemini] add unitests to check gemini correctness (#2015)	2 years ago