公司名称logo设计,江门网站优化快速排名,深圳龙华区属于什么风险区,东莞网站建设方案企业定义 提供了给定数据集的迭代器
torch.utils.data.DataLoader(dataset,
batch_size1, 每次拿多少数据
shuffleNone, 是否打乱
samplerNone,
batch_samplerNone,
num_workers0, 多进程#xff08;加载数据时采用#xff09;默认是0,使用主进程加载数据
collate_fnNone,
p…定义 提供了给定数据集的迭代器
torch.utils.data.DataLoader(dataset,
batch_size1, 每次拿多少数据
shuffleNone, 是否打乱
samplerNone,
batch_samplerNone,
num_workers0, 多进程加载数据时采用默认是0,使用主进程加载数据
collate_fnNone,
pin_memoryFalse,
drop_lastFalse, 总数据/batch_size之后的余数是舍去True还是不舍去(False)
timeout0,
worker_init_fnNone, multiprocessing_contextNone, generatorNone, *, prefetch_factorNone, persistent_workersFalse, pin_memory_device)官方描述 2. 使用
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
import torchvisiondataset_transform torchvision.transforms.Compose([torchvision.transforms.ToTensor()
])
# 测试数据集
test_set torchvision.datasets.CIFAR10(root./dataset,transformdataset_transform,trainFalse,downloadTrue)
# 测试集中的第一张图片及target
img,target test_set[0]
print(img.shape,target)
# 在终端中查看tensorboard --logdirdataloader
writer SummaryWriter(dataloader)
test_loader DataLoader(datasettest_set,batch_size64,shuffleTrue,num_workers0,drop_lastTrue)
for epoch in range(2):step 0for data in test_loader:imgs,targets data# print(imgs.shape)# print(targets)writer.add_images(Epoch:{}.format(epoch),imgs,step)step 1
writer.close()
