官术网_书友最值得收藏!

PyTorch dataset loaders

Pytorch includes data loaders for several datasets to help you get started. The torch.dataloader is the class used for loading datasets. The following is a list of the included torch datasets and a brief description:

Here is a typical example of how we load one of these datasets into PyTorch:

CIFAR10 is a torch.utils.dataset object. Here, we are passing it four arguments. We specify a root directory relative to where the code is running, a Boolean, train, indicating if we want the test or training set loaded, a Boolean that, if set to True, will check to see if the dataset has previously been downloaded and if not download it, and a callable transform. In this case, the transform we select is ToTensor(). This is an inbuilt class of torchvision.transforms that makes the class return a tensor. We will discuss transforms in more detail later in the chapter.

The contents of the dataset can be retrieved by a simple index lookup. We can also check the length of the entire dataset with the len function. We can also loop through the dataset in order. The following code demonstrates this:

主站蜘蛛池模板: 分宜县| 平陆县| 肃南| 巫溪县| 建德市| 萍乡市| 昌图县| 同仁县| 资源县| 年辖:市辖区| 安达市| 绥阳县| 洛阳市| 利辛县| 隆昌县| 馆陶县| 上蔡县| 自治县| 新疆| 麻阳| 镇康县| 高密市| 措美县| 临西县| 克东县| 绥宁县| 牟定县| 长寿区| 汝城县| 岳阳县| 台湾省| 武川县| 新竹市| 岳普湖县| 昆山市| 连江县| 阜新市| 北京市| 且末县| 沂水县| 抚远县|