ColossalAI/tests/test_tensor/model
Jiarui Fang 2e9cbfca12
[Gemini] add unitests to check gemini correctness (#2015)
2022-11-24 16:51:45 +08:00
..
test_gpt2.py [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00
test_model.py [utils] remove lazy_memory_allocate from ColoInitContext (#1844) 2022-11-09 11:50:33 +08:00
test_module_spec.py [utils] remove lazy_memory_allocate from ColoInitContext (#1844) 2022-11-09 11:50:33 +08:00