ColossalAI/tests/components_to_test
Frank Lee 526a318032 [unit test] Refactored test cases with component func (#339)
* refactored test with component func

* fixed bug
2022-03-11 15:50:28 +08:00
..
utils [unit test] Refactored test cases with component func (#339) 2022-03-11 15:50:28 +08:00
__init__.py add bert for unitest and sharded model is not able to pass the bert case 2022-03-11 15:50:28 +08:00
bert.py fix bert unit test 2022-03-11 15:50:28 +08:00
nested_model.py [unit test] Refactored test cases with component func (#339) 2022-03-11 15:50:28 +08:00
registry.py [test] refactored testing components (#324) 2022-03-11 15:50:28 +08:00
repeated_computed_layer.py [zero] update zero context init with the updated test utils (#327) 2022-03-11 15:50:28 +08:00
resnet.py [zero] update zero context init with the updated test utils (#327) 2022-03-11 15:50:28 +08:00