深度学习——DatadLoader的使用

一、DataLoader简介#

官网地址:

torch.utils.data — PyTorch 2.0 documentation

1. DataLoder类#

Copy

class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=None, sampler=None, batch_sampler=None, num_workers=0, collate_fn=None, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None, multiprocessing_context=None, generator=None, *, prefetch_factor=None, persistent_workers=False, pin_memory_device='')

由此可见，DataLoder必须需要输入的参数只有�������。

2. 参数说明#

• dataset(Dataset): 数据集的储存的路径位置等信息

• batch_size(int): 每次取数据的数量，比如batchi_size=2，那么每次取2条数据</