pytorch中的Dataset和DataLoader

Pytorch有两个读取数据的方式:
使用Dataset
使用DataLoader
本文先介绍第一种——Dataset

Dataset与DataLoader区别
Dataset:提供一种方法,去获取数据及其对应的label值
DataLoader:提供一种方法,可以以特定的形式打包数据

Reference

  1. Pytorch Tutorials > Datasets & DataLoaders
  2. PyTorch DataLoader二三事(上)

【Pytorch学习】-- 读取数据 -- torch.utils.data -- Dataset

posted @ 2024-03-19 11:11  光辉233  阅读(11)  评论(0编辑  收藏  举报