1. Learn
  2. /
  3. 课程
  4. /
  5. 使用 PyTorch 的深度学习入门

Connected

道练习

使用 DataLoader

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

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

说明

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