pytorch如何加载MNIST数据集

最新推荐文章于 2024-10-17 17:34:08 发布

FR52

最新推荐文章于 2024-10-17 17:34:08 发布

阅读量612

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/Mtf007/article/details/128287537

版权

CIFAR10和CIFAR100类似

import torchvision
import torch
import torchvision.transforms as transforms
batch_size = 256
train_dataset = torchvision.datasets.MNIST(root='../../data',
                                           train=True,
                                           transform=transforms.ToTensor(),
                                           download=True)

test_dataset = torchvision.datasets.MNIST(root='../../data',
                                          train=False,
                                          transform=transforms.ToTensor())

# Data loader
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
                                           batch_size=batch_size,
                                           shuffle=True)

test_loader = torch.utils.data.DataLoader(dataset=test_dataset,
                                          batch_size=batch_size,
                                          shuffle=False)