basecls.data.dataloader#
- basecls.data.dataloader.build_dataloader(cfg, train, augments=None, mode='folder', infinite=False, rank=None)[源代码]#
Build function for MegEngine dataloader.
- 参数
cfg (
ConfigDict
) – config for building dataloader.train (
bool
) – train set or test set.augments (
Optional
[Transform
]) – augments for building dataloder. Default:None
infinite (
bool
) – make dataloader infinite or not. default:False
rank (
Optional
[int
]) – machine rank, only useful for infinite dataloader. Default:None
- 返回类型
- 返回
A dataloader.