RuntimeError:The size of tensor a (128) must match the size of tensor b (256) at non-singleton dime

赶快溜

已于 2024-10-26 18:45:40 修改

阅读量5.3k

点赞数 2

分类专栏： pytorch报错算法文章标签：计算机视觉

于 2023-05-10 14:37:41 首次发布

本文链接：https://blog.youkuaiyun.com/m0_45924886/article/details/130597462

版权

14 篇文章

订阅专栏

2 篇文章

订阅专栏

文章讲述了在计算损失函数时遇到的问题，即output和target的尺寸不一致。为了解决这个问题，建议检查模型的输出尺寸，可以使用torchstat库来统计模型的输出信息。示例中展示了如何初始化InpaintGenerator模型并打印其输出尺寸，以确保与输入图片尺寸匹配。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

报错如上，是因为在计算损失的时候，output和target的尺寸大小不一样，所以修改的方法是查看模型参数，也就是查看output的输出大小是多少，可以使用一下方法：

from torchstat import stat

model=InpaintGenerator(1)  #1是InpaintGenerator的参数
print(model)
print(stat(model,input_size=[4,256,256]))  #input_size是输入图片的大小