torch.utils.data.DataLoader(
时间: 2023-10-16 09:06:54 浏览: 59
是用来加载数据的Python包torch.utils.data中的一个类。它提供了一个可迭代的数据加载器,用于批处理和并行加载数据。通过DataLoader,您可以将自定义的数据集对象或torchvision中的预定义数据集(如MNIST、CIFAR等)传递给模型进行训练或推理。
DataLoader可以设置多个参数,包括数据集对象、批处理大小、是否进行随机洗牌、使用多进程加载数据等。它会自动将数据集分成小批量,并在每个批次中并行加载和预处理数据,以提高训练效率。
以下是一个示例用法:
```python
import torch
from torch.utils.data import DataLoader
from torchvision.datasets import MNIST
from torchvision.transforms import ToTensor
# 创建MNIST数据集对象
dataset = MNIST(root='data/', train=True, transform=ToTensor())
# 创建数据加载器
dataloader = DataLoader(dataset, batch_size=64, shuffle=True)
# 迭代数据加载器
for images, labels in dataloader:
# 在这里进行模型训练或推理
pass
```
在上面的示例中,我们首先创建了一个MNIST数据集对象,并传递给DataLoader作为参数。然后设置了批处理大小为64,并启用了随机洗牌。最后,我们使用for循环迭代DataLoader,每次获取一个批次的图像和标签数据用于模型的训练或推理。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![ipynb](https://img-home.csdnimg.cn/images/20210720083646.png)
![py](https://img-home.csdnimg.cn/images/20210720083646.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)