三、dataloader的使用

使用dataloader可以批量加载数据

1. 导入dataloader:

from torch.utils.data import DataLoader

2. dataloader的使用方法：
   
   根据上图我们可以知道dataloader所需要的参数，包括dataset、batch_size、shuffle、sampler、num_workers等

3. 导入一个数据集（这里直接使用的是torch内嵌的数据集)

import torchvision
# 准备测试集
test_data=torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor())

4. dataloader的使用：

# 加载数据集
test_loader=DataLoader(dataset=test_data,batchz_size=4,shuffle=True,num_workers=0,drop_last=False)

# 将每一个轮次都写进writer里，从而方便对比
for epoch in range(2):
	step=0
	for data in data_loader:
		imgs,targets=data
		writer.add_images("Epoch:{}".format(epoch),imgs,step)
		step+=1