![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
yezhangyinge
这个作者很懒,什么都没留下…
展开
-
ValueError: expected 2D or 3D input (got 4D input)
通过查看代码找到了问题所在是因为使用了不正确的BatchNorm函数原因分析我们可以看到官方文档对于BatchNorm1d的解释BatchNorm1d显然需要输入的是二维或者三维,但是我们有一个四维的输入那么我们可以查看BatchNorm2d的官方解释BatchNorm2d可以接受四维输入,解决方法根据上述分析,把BatchNorm1d改为BatchNorm2d即可...原创 2021-05-01 17:12:37 · 14807 阅读 · 2 评论 -
Error: Expected more than 1 value per channel when training
Error: Expected more than 1 value per channel when training问题描述:在深度学习网络里使用了BatchNorm,但是出现输入的batch输入大小为1时候会出现这个错误原因分析:batchNorm,要求输入batch>1,所以会出现这个错误解决方案:方法一:只需要在test的时候加入network.eval()方法二:在dataloader里加上drop_last=True...原创 2021-04-21 23:15:40 · 208 阅读 · 0 评论