pytorch中的dataset和datalaoder

1. 使用Dataset数据集

import torchvision

使用方法：

# 1. 如何使用dataset标准数据集
train_set = torchvision.datasets.CIFAR10(root="./dataset2", train=True, transform=dataset_transform, download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset2", train=False, transform=dataset_transform, download=True)
# 获取img和target
img, target = test_set[i]

DataLoader

from torch.utils.data import DataLoader
常用参数：
dataset
batch_size：每次取数据集的大小
shuffle：每次循环取是否打乱。True为打乱（常用），False为不打乱。
num_workers
drop_last：剩余的数据集是否保留。True为保留，False为不保留

test_loader = DataLoader(dataset=test_data, batch_size=64, shuffle=True, num_workers=0, drop_last=False)