github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-18 21:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity
0df5034a36
ColossalAI/tests/components_to_test/__init__.py
ver217 8e3d0ad8f1
[unit test] refactor test tensor (#1005) 
* polish test_gpt

* update op unit tests

* update test model
2022-05-19 18:57:56 +08:00

2 lines
100 B
Python
Raw Blame History

from . import repeated_computed_layer, resnet, nested_model, bert, no_leaf_module, simple_net, gpt
Reference in New Issue View Git Blame Copy Permalink