AttributeError: module 'torchvision.datasets.mnist' has no attribute 'test_images'
时间: 2023-12-01 16:42:43 浏览: 195
测试图片资源选择问题
这个错误通常是因为您正在使用的torchvision版本不兼容。在较新的torchvision版本中,test_images已被更改为test_data。您可以尝试更新torchvision以解决此问题。如果更新不起作用,您可以尝试使用以下代码来加载MNIST数据集:
```python
import torch
from torchvision import datasets, transforms
transform = transforms.Compose([transforms.ToTensor(),
transforms.Normalize((0.5,), (0.5,)),
])
trainset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=True, transform=transform)
testset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=False, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)
testloader = torch.utils.data.DataLoader(testset, batch_size=64, shuffle=True)
images, labels = next(iter(testloader))
print(images.shape)
```
阅读全文