ColossalAI/colossalai/nn/data/sampler/__init__.py

5 lines
143 B
Python

from .base_sampler import BaseSampler
from .data_parallel_sampler import DataParallelSampler
__all__ = ['BaseSampler', 'DataParallelSampler']