2021
09-24
09-24
手把手教你实现PyTorch的MNIST数据集
目录概述获取数据网络模型train函数test函数main函数完整代码:概述MNIST包含0~9的手写数字,共有60000个训练集和10000个测试集.数据的格式为单通道28*28的灰度图.获取数据defget_data():"""获取数据"""#获取测试集train=torchvision.datasets.MNIST(root="./data",train=True,download=True,transform=torchvision.transforms.Compose([...
继续阅读 >