Files
ColossalAI/colossalai/nn/data/sampler/__init__.py
2021-10-28 18:21:23 +02:00

5 lines
143 B
Python

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