ColossalAI/tests/components_to_test/__init__.py

2 lines
61 B
Python

from . import repeated_computed_layer, resnet, nested_model