import torch
from torch.utils.data import DataLoader
import torchvision.datasets as dsets
import torchvision.transforms as transforms
batch_size=100
train_dataset = dsets.MNIST(root='/pymnist',
train=True,
transform=transforms.ToTensor(),
download=True)
test_dataset = dsets.MNIST(root='/pymnist',
train=False,
transform=transforms.ToTensor(),
download=True)
train_loader=torch.utils.data.DataLoader(dataset=train_dataset,
batch_size=batch_size,
shuffle=True)
test_loader=torch.utils.data.DataLoader(dataset=train_dataset,
batch_size=batch_size,
运用pytorch实现MNIST分类(含完整代码)
最新推荐文章于 2025-02-26 21:31:41 发布