ColossalAI/tests/components_to_test/utils/__init__.py

3 lines
87 B
Python

from .dummy_data_generator import DummyDataGenerator
from .executor import run_fwd_bwd