今天租了一个GPU,出现一些情况,记录一下 因为jupyter notebook的一些原因,无法直接从网络上导入Fashion-MNIST数据,故而手工导入. Python读取本地Fashion-MNIST数据集 我的截图: 把FashionMNIST文件夹放到名称为’wode’的文件夹里面 如果不改变图片大小:
trans=[]
trans.append(torchvision.transforms.ToTensor())
transform = torchvision.transforms.Compose(trans)
mnist_train = torchvision.datasets.FashionMNIST(root=r'wode/',train=True, transform=transform, download=False)
mnist_test = torchvision.datasets.FashionMNIST(root=r'wode/',train=False, transform=transform, download=False)
len(mnist_train),len(mnist_test)
还有如果改变图片大小224*224:
trans=[]
trans.append(torchvision.transforms.Resize(size=224))
trans.append(torchvision.transforms.ToTensor())
transform = torchvision.transforms.Compose(trans)
mnist_train = torchvision.datasets.FashionMNIST(root=r'wode/',train=True, transform=transform, download=False)
mnist_test = torchvision.datasets.FashionMNIST(root=r'wode/',train=False, transform=transform, download=False)
len(mnist_train),len(mnist_test)
而后batch_size自定,将训练集和测试集分化为批量迭代数据集
batch_size=256
train_iter=data.DataLoader(mnist_train,batch_size,shuffle=True,num_workers=4)
test_iter=data.DataLoader(mnist_test,batch_size,shuffle=True,num_workers=4)
|