StreamSampler

class StreamSampler(batch_size=1)

从流数据集中采样。

警告

在多台机器的情况下，采样器应确保每个 worker获取不同的数据。但是此类还无法做到这一点，请构建您自己的数据集和采样器以达成此目标。

通常情况下， __iter__ 可以通过 rank = dist.get_rank() 返回不同的迭代器，用于获取不同数据。

上一页

MapSampler

下一页

SequentialSampler