微信网站开发工具,网站运维托管,广州网站设计平台,wordpress 主题域名授权一、DataLoader参数解析
DataLoader官网使用手册
参数描述dataset说明数据集所在的位置、数据总数等batch_size每次取多少张图片shuffleTrue乱序、False顺序(默认)samplerbatch_samplernum_workers多进程#xff0c;默认为0采用主进程加载数据collate_fnpin_memorydrop_las…一、DataLoader参数解析
DataLoader官网使用手册
参数描述dataset说明数据集所在的位置、数据总数等batch_size每次取多少张图片shuffleTrue乱序、False顺序(默认)samplerbatch_samplernum_workers多进程默认为0采用主进程加载数据collate_fnpin_memorydrop_lastTrue最后不满一个batch_size的去掉False最后剩余多少也都要timeoutworker_init_fngeneratorprefetch_factorprefetch_factorpin_memory_device
import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWritertest_data torchvision.datasets.CIFAR10(CIFAR_10,trainFalse,transformtorchvision.transforms.ToTensor())
#Ctrl按住点击CIFAR10找到__getitem__类查看返回值return img, targettest_loader DataLoader(datasettest_data,batch_size64,shuffleTrue,num_workers0,drop_lastFalse)
#drop_lastFalse最后剩余多少也都要
#shuffleTrue乱序#测试一下
img,target test_data[0]
print(img.shape)#torch.Size([3, 32, 32])
print(target)#3writer SummaryWriter(y_log)
i 0
for data in test_loader:imgs,targets data#print(type(imgs))writer.add_images(test_data,imgs,i)i i1writer.close()在Terminal下运行tensorboard --logdiry_log --port2312logdir为打开事件文件的路径port为指定端口打开 通过指定端口2312进行打开tensorboard若不设置port参数默认通过6006端口进行打开。 点击该链接或者复制链接到浏览器打开即可 import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWritertest_data torchvision.datasets.CIFAR10(CIFAR_10,trainFalse,transformtorchvision.transforms.ToTensor())
#Ctrl按住点击CIFAR10找到__getitem__类查看返回值return img, targettest_loader DataLoader(datasettest_data,batch_size64,shuffleFalse,num_workers0,drop_lastTrue)
#drop_lastTrue不满一组的舍去
#shuffleFalse顺序img,target test_data[0]
print(img.shape)#torch.Size([3, 32, 32])
print(target)#3writer SummaryWriter(y_log)for epoch in range(3):i 0for data in test_loader:imgs,targets data#print(type(imgs))writer.add_images(Epoch:{}.format(epoch),imgs,i)i i1writer.close()