参考链接:解决Pytorch dataloader时报错每个tensor维度不一样的问题_python_脚本之家
记录一下自己遇到的bug:
问题描述:
问题分析:
torch.stack(batch, 0, out=out)出错,原因可能是:同一个batch的数据图片的维度(H, W, C)要相同(可以见官方文档:其shape必须一致)
问题解决:文章来源:https://www.toymoban.com/news/detail-610603.html
如果图像尺寸不固定一致的话,那batch_size设置为1,问题即可解决文章来源地址https://www.toymoban.com/news/detail-610603.html
到了这里,关于RuntimeError: stack expects each tensor to be equal size, but got at entry的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!