如何使用pytorch下载MNIST数据集

编程学习中09

于 2021-07-05 08:40:25 发布

阅读量4.3k

点赞数 5

文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/lichengjie1/article/details/118479758

版权

代码如下：

# 1.定义超参数
BATCH_SIZE = 16

# 2.构建pipeline，对图片做一些变换
pipeline = transforms.Compose([
    transforms.ToTensor(),     
    transforms.Normalize((0.1307,),(0.3081,)) 
])

# 3.加载数据集
from torch.utils.data import DataLoader

# 下载数据集
train_set = datasets.MNIST("data",train=True,download=True,transform=pipeline)

test_set = datasets.MNIST("data",train=False,download=True,transform=pipeline)

# 加载数据
train_loader = DataLoader(train_set,batch_size=BATCH_SIZE,shuffle=True)
test_loader = DataLoader(test_set,batch_size=BATCH_SIZE,shuffle=True)