深度学习模型tensor维度对不上是一个非常常见且有时比较难排查的现象。之所以难排查是因为报错信息和真实的错误原因之间的联系往往并不紧密，很难仅仅从PyTorch给出的数字上的信息判断错误在何处。
笔者在一个训练好的模型上测试单个样本时出现了这个问题，一度排查了图片读取、图片通道数等问题，最后发现原因在于forward部分的代码是针对batch编写的，所以输入的格式是四维的BxCxHxW，而测试单个样本时输入是三维的，所以只需要一行img=img.unsqueeze(0)就解决了这个浪费了一上午 + 一下午的问题。。。
本文发布于:2024-02-03 03:20:50，感谢您对本站的认可！
本文链接：https://www.4u4v.net/it/170690165148326.html
版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系，我们将在24小时内删除。
| 留言与评论（共有 0 条评论） |
