参考链接：解决Pytorch dataloader时报错每个tensor维度不一样的问题_python_脚本之家
记录一下自己遇到的bug：
问题描述：
问题分析：
torch.stack(batch, 0, out=out)出错，原因可能是：同一个batch的数据图片的维度（H, W, C）要相同（可以见官方文档：其shape必须一致）
问题解决：
如果图像尺寸不固定一致的话，那batch_size设置为1，问题即可解决
文章出处登录后可见！
已经登录？立即刷新
赞 (0)
Python 操作 Excel 全攻略 | 包括读取、写入、表格操作、图像输出和字体设置
上一篇
2023年7月15日
pycharm找不到conda可执行文件怎么办？
下一篇
2023年7月15日