开始使用免费开始使用

使用 DataLoader

DataLoader 类对于高效处理大型数据集至关重要。它能加速训练、优化内存使用,并使梯度更新更稳定,从而让深度学习模型更有效。

现在,您将使用上一个练习中的 dataset 创建一个 PyTorch DataLoader,并查看其运行效果。

本练习是课程的一部分

使用 PyTorch 的深度学习入门

查看课程

练习说明

  • 导入所需模块。
  • 使用 dataset 创建一个 DataLoader,将批大小设为 2,并启用随机打乱。
  • 遍历该 DataLoader,并打印每个批次的输入和标签。

交互式实操练习

通过完成这段示例代码来试试这个练习。

from torch.utils.data import ____

# Create a DataLoader
dataloader = ____

# Iterate over the dataloader
for batch_inputs, batch_labels in dataloader:
    print('batch_inputs:', batch_inputs)
    print('batch_labels:', batch_labels)
编辑并运行代码