DataLoader的使用1

dataset

Torchvision provides many built-in datasets in the torchvision.datasets module, as well as utility classes for building your own datasets.

使用方法

All the datasets have almost similar API. They all have two common arguments: transform and target_transform to transform the input and target respectively. You can also create your own datasets using the provided base classes.

dataloader

loader：加载器。把东西加载到神经网络中

作用

从dataset中取数据

• 取多少
• 取什么

官网学习dataloader

在官网中找到dataloader

学习常见参数意义的学习

dataset (Dataset) – dataset from which to load the data.

batch_size (int, optional) – how many samples per batch to load (default: 1).

shuffle (bool, optional) – set to True to have the data reshuffled at every epoch (default: False).

sampler (Sampler or Iterable, optional) – defines the strategy to draw samples from the dataset. Can be any Iterable with __len__ implemented. If specified, shuffle must not be specified.

drop_last (bool, optional) – set to True to drop the last incomplete batch, if the dataset size is not divisible by the batch size. If False and the size of dataset is not divisible by the batch size, then the last batch will be smaller. (default: False)
1
2
3
4
5
6
7
8
9

---

其他

专业英语

• batch一批，一批生产量
• shuffle 重新洗牌