RuntimeError: stack expects each tensor to be equal size, but got at entry

• 人工智能

参考链接：​​​​​​解决Pytorch dataloader时报错每个tensor维度不一样的问题_python_脚本之家

记录一下自己遇到的bug：

问题描述： 

RuntimeError: stack expects each tensor to be equal size, but got at entry问题分析：

torch.stack(batch, 0, out=out)出错，原因可能是：同一个batch的数据图片的维度（H, W, C）要相同（可以见官方文档：其shape必须一致）

问题解决：

如果图像尺寸不固定一致的话，那batch_size设置为1，问题即可解决

本文来自网络，不代表协通编程立场，如若转载，请注明出处：https://www.net2asp.com/7833f9da62.html

赞 (0)
« 上一篇 1天前
下一篇 » 1天前

相关推荐