megengine.data.StreamSampler¶

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

从流数据集中采样。

警告

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

Usually, __iter__ can return different iterator by rank = dist.get_rank(). So that they will get different data.

方法

megengine.data.MapSampler.scatter

megengine.data.SequentialSampler