Original Traceback (most recent call last): File "D:\anaconda3\.conda\envs\DA\lib\site-packages\torch\utils\data\_utils\worker.py", line 202, in _worker_loop data = fetcher.fetch(index) File "D:\anaconda3\.conda\envs\DA\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "D:\anaconda3\.conda\envs\DA\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "D:\anaconda3\.conda\envs\DA\lib\site-packages\torchvision\datasets\mnist.py", line 109, in __getitem__ img = Image.fromarray(img.numpy(), mode='L') AttributeError: 'numpy.ndarray' object has no attribute 'numpy' 进程已结束,退出代码1
时间: 2023-06-14 09:08:27 浏览: 192
这个错误是因为在使用 torchvision 的 MNIST 数据集时,返回的数据类型是 numpy 数组,而在使用 Image.fromarray 转换时,需要传入的是 Tensor 类型的数据,因此需要将 numpy 数组转换为 Tensor 类型的数据。可以使用 `torch.from_numpy()` 方法将 numpy 数组转换为 Tensor 类型的数据,例如:
```
img = Image.fromarray(torch.from_numpy(img).numpy(), mode='L')
```
这样就可以避免这个错误了。
阅读全文