pytorch中DataLoader函数

原创

于 2019-12-19 16:39:29 发布 · 4.6k 阅读

35 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习

PyTorch的DataLoader用于高效地加载数据集，它结合了数据集和采样器，提供单进程或多进程迭代。DataLoader接受参数如dataset、batch_size、shuffle等，可以自定义sampler、collate_fn等行为。在使用时，DataLoader返回的迭代器可以用iter()获取，通过next()或for循环遍历，同时，collate_fn可以定制样本拼接规则。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, num_workers=0, collate_fn=<function default_collate>, pin_memory

最低0.47元/天解锁文章