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

返回类型

DataLoader

返回

A dataloader.