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

github/ColossalAI

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

Files

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

2022-03-09 13:30:38 +08:00

2 lines

67 B

Python

Raw Blame History

from . import repeated_computed_layer, resnet, nested_model, bert