import torchvision
from torch.utils.data import DataLoader
#这里使用的公用数据集CIFAR10
test_data = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor(),download=True)
#将数据分成多个batch。drop_last当剩余数据不足以组成一个batch时删除这些数据。shuffle为每轮训练划分batch时划分结果是否一样
test_loader = DataLoader(dataset=test_data,batch_size=3,shuffle=True,drop_last=True)
features,target = test_data[0]
print(features.shape," ",target)
DataLoader划分Batch
最新推荐文章于 2024-08-16 15:24:54 发布