ColossalAI/tests/components_to_test
ver217 f5f0ad266e fix bert unit test 2022-03-11 15:50:28 +08:00
..
utils [test] refactored testing components (#324) 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 [zero] update zero context init with the updated test utils (#327) 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