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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 22:54:10 +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