DataLoader是使用Pytorch框架进行数据加载时使用到的一个库。训练数据和测试数据均需通过DataLoader加载后方可被网络使用。
class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False,
sampler=None, num_workers=0, collate_fn=<function default_collate>, pin_memory=False, drop_last=False)
## dataset(Dataset): 加载数据的数据集
## batch_size(int, optional): 每个batch加载多少个样本数据(默认:1)。
## shuffle(bool, optional): 设置True时会在每个epoch重新打乱数据顺序(默认:False)。
## sampler(Sampler, optional): 定义从数据集中提取样本的策略。如果指定,则忽略shuffle参数。
## num_workers(int, optional): 加载数据的子进程数。0表示数据将在主进程中加载(默认:0)。
## collate_fn(callable, optional):
## pin_memory(bool, optional):
## drop_last(bool, optional): 如果数据集大小不能被batch size整除,