如何解决IndexError:尺寸超出范围预计在[-1,0]范围内,但得到2
plt.imshow(torchvision.utils.make_grid(images[3],nrow=5).permute(1,2,0 )
为什么在运行上述代码时弹出以下错误?
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-78-ce90d0e2159a> in <module>
----> 1 plt.imshow(torchvision.utils.make_grid(images[3],0))
~\Anaconda3\lib\site-packages\torchvision\utils.py in make_grid(tensor,nrow,padding,normalize,range,scale_each,pad_value)
74 xmaps = min(nrow,nmaps)
75 ymaps = int(math.ceil(float(nmaps) / xmaps))
---> 76 height,width = int(tensor.size(2) + padding),int(tensor.size(3) + padding)
77 num_channels = tensor.size(1)
78 grid = tensor.new_full((num_channels,height * ymaps + padding,width * xmaps + padding),pad_value)
IndexError: Dimension out of range (expected to be in range of [-1,0],but got 2)
这是在运行以下代码后发生的
model= nn.Sequential(nn.Linear (150528,1000),nn.ReLU(),nn.Linear(1000,250),nn.Linear(250,32),nn.LogSoftmax(dim=1),)
criterion = nn.CrossEntropyLoss()
images = images.view(images.shape[0],-1)
logits = model (images)
loss = criterion(logits,labels)
print(loss)
然后重新运行上面的代码
plt.imshow(torchvision.utils.make_grid(images[3],0 )
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。