ColossalAI/tests/components_to_test
Jiarui Fang eb7742a4bb
[Gemini] more tests for Gemini (#2038)
* [Gemini] more tests for Gemini

* polish code
2022-11-29 17:13:10 +08:00
..
utils [Gemini] more rigorous unit tests for run_fwd_bwd (#2034) 2022-11-29 09:26:06 +08:00
__init__.py [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00
bert.py [Gemini] more tests for Gemini (#2038) 2022-11-29 17:13:10 +08:00
gpt.py [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00
inline_op_model.py [testing] fix testing models (#2036) 2022-11-29 13:42:06 +08:00
nested_model.py [testing] fix testing models (#2036) 2022-11-29 13:42:06 +08:00
no_leaf_module.py [testing] fix testing models (#2036) 2022-11-29 13:42:06 +08:00
registry.py [test] refactored testing components (#324) 2022-03-11 15:50:28 +08:00
repeated_computed_layer.py [testing] fix testing models (#2036) 2022-11-29 13:42:06 +08:00
resnet.py [zero] new interface for ShardedOptimv2 (#406) 2022-03-14 20:48:41 +08:00
simple_net.py [testing] fix testing models (#2036) 2022-11-29 13:42:06 +08:00