megengine.data.DataLoader

class DataLoader(dataset, sampler=None, transform=None, collator=None, num_workers=0, timeout=0, timeout_event=<function raise_timeout_error>, divide=False)[源代码]

基类:object

可用于在给定数据集上遍历并取得数据。

方法

__init__(dataset[, sampler, transform, …])

DataLoader 将数据集与sampler 、transform 、 collator 结合起来,从而能够灵活地从数据集中连续获得拼成 minibatch 的数据。