ColossalAI/tests/components_to_test/__init__.py at e79ea44247bbb6457cd2a2c30454208017fd31ef - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 23:22:42 +00:00

Files

jiaruifang 7977422aeb add bert for unitest and sharded model is not able to pass the bert case

2022-03-11 15:50:28 +08:00

2 lines

67 B

Python

Raw Blame History

from . import repeated_computed_layer, resnet, nested_model, bert