AttributeError: 'MNIST' object has no attribute 'to'
时间: 2023-12-10 10:36:03 浏览: 221
AttributeError: ‘NoneType’ object has no attribute ‘children’ 错误
5星 · 资源好评率100%
AttributeError: 'MNIST' object has no attribute 'to'这个错误通常是因为在使用PyTorch的MNIST数据集时,将其当作了Pandas的DataFrame或Series对象来处理,而这些对象没有to()方法。解决这个问题的方法是将MNIST数据集转换为PyTorch的Tensor对象,可以使用torchvision.transforms.ToTensor()方法来实现。具体的代码如下所示:
```python
import torchvision.transforms as transforms
from torchvision.datasets import MNIST
# 加载MNIST数据集
train_dataset = MNIST(root='./data', train=True, transform=transforms.ToTensor(), download=True)
test_dataset = MNIST(root='./data', train=False, transform=transforms.ToTensor())
# 对数据集进行处理
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)
```
阅读全文