github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/components_to_test/__init__.py

2 lines
67 B
Python
Raw Normal View History

add bert for unitest and sharded model is not able to pass the bert case
2022-03-09 02:39:02 +00:00
from . import repeated_computed_layer, resnet, nested_model, bert