megengine.data.StreamSampler¶

class StreamSampler(batch_size=1)[源代码]¶

基类：megengine.data.sampler.Sampler

从流数据集中采样。

警告

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

通常，〜.StreamDataset.__ iter__ 可以通过 rank = dist.get_rank（) 返回不同的迭代器，这样他们将获得不同的数据。

方法

__init__([batch_size])

初始化方法。