ColossalAI/tests/components_to_test/__init__.py

2 lines
100 B
Python
Raw Normal View History

from . import repeated_computed_layer, resnet, nested_model, bert, no_leaf_module, simple_net, gpt