在运行pytorch写的网络的时候遇到了一些问题

Question

RuntimeError: Traceback (most recent call last):
  File "G:\anaconda\lib\site-packages\torch\utils\data\dataloader.py", line 138, in _worker_loop
    samples = collate_fn([dataset[i] for i in batch_indices])
  File "G:\anaconda\lib\site-packages\torch\utils\data\dataloader.py", line 138, in <listcomp>
    samples = collate_fn([dataset[i] for i in batch_indices])
  File "G:\anaconda\lib\site-packages\torchvision\datasets\mnist.py", line 95, in __getitem__
    img = self.transform(img)
  File "G:\anaconda\lib\site-packages\torchvision\transforms\transforms.py", line 60, in __call__
    img = t(img)
  File "G:\anaconda\lib\site-packages\torchvision\transforms\transforms.py", line 163, in __call__
    return F.normalize(tensor, self.mean, self.std, self.inplace)
  File "G:\anaconda\lib\site-packages\torchvision\transforms\functional.py", line 208, in normalize
    tensor.sub_(mean[:, None, None]).div_(std[:, None, None])
RuntimeError: output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]

请问这个问题要怎么解决啊？
这些文件都是安装包自带的，我也没改过啊x

Answer 1

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 这篇文章讲的很详细，请看：记录：运行pytorch时遇到的一些报错

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 以帮助更多的人 ^-^