ColossalAI/tests/test_gemini
Jiarui Fang 978242326a
[Gemini] remove eval in gemini unittests! (#2092)
2022-12-07 11:58:37 +08:00
..
update [Gemini] remove eval in gemini unittests! (#2092) 2022-12-07 11:58:37 +08:00
test_gemini_manager.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
test_param_op.py [test] align model name with the file name. (#2045) 2022-11-30 15:45:26 +08:00
test_runtime_mem_tracer.py [Gemini] use MemStats in Runtime Memory tracer (#2088) 2022-12-06 19:48:20 +08:00