pytorch学习笔记------batchsize制作（分类任务）

基于pytorch框架训练batchsize制作

from torchvision import datasets
import os
import torch
batch_size=8#训练batchsize
data_dir=''#数据集文件夹
image_datasets={x:datasets.ImageFolder(os.path.join(data_dir,x),data_transforms[x]) for x in ['train','valid']}#dataset
dataloaders={x:torch.utils.data.DataLoader(image_datasets[x],batch_size=batch_size,shuffle=True) for x in ['train','valid'] }#dataloader加载
data_sizes={x:len(image_datasets[x])for x in ['train','valid']}#数据大小
class_name=image_datasets['train'].classes#类名