ColossalAI/tests/test_gemini
HELSON a3100bd50d
[testing] add beit model for unit testings (#2196)
* [testing] add beit model

* [beit] fix bugs

* [beit] fix bugs

* [testing] fix bugs
2022-12-26 17:35:36 +08:00
..
update [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_gemini_manager.py
test_param_op.py
test_runtime_mem_tracer.py [gemini] get the param visited order during runtime (#2108) 2022-12-09 16:13:03 +08:00