PyTorch报错shape ‘[16, 1, 28, 28]‘ is invalid for input of size 6272?尝试设置-1

枫

debug

    • 食用指南:
    • 框架/语言:
    • 问题描述:
    • 解决方案:
      • 问题原因:
      • 解决方案:
    • 安利时间:

食用指南:

框架使用过程中的问题首先要注意版本发布时间造成方法弃用
当你在CSDN等网站查找不到最新版本框架的问题解决,别迟疑,马上去Goole
语言问题比框架问题好解决
如果本篇debug博客解决了你的问题,不妨留下你的👍
欢迎关注博主,查看更多c/c++/JAVA/算法/框架知识

框架/语言:

  • PyTorch

问题描述:

  • 使用tensor.reshape() 或 torch.reshape(data, (dim1, dim2, …)) 规范矩阵类型时,报错:
    shape ‘[16, 1, 28, 28]’ is invalid for input of size 6272
  • 错误场景:
    reshape error

解决方案:

问题原因:

  • reshape参数过于详细:
    默认每次加载的矩阵都是16*1*28*28
data = data.reshape(16, 1, 28, 28)
# 第一个参数是batch_size
# 第二个参数的通道数目,灰度图一般为1,彩色图为RGB三通道
# 之后的参数是图像的高度和宽度(可能还有深度)
  • dataloader可能有余数:
    batch_size表示每次取和同时处理的图像张数,但是可能图像总数并非正好是batch_size倍数

解决方案:

  • reshape(-1, …):
    最后一次迭代dataloader时,可能不足batch_size张图片,
    这时参数设为-1,表示有多少取多少即可
data = data.reshape(16, 1, 28, 28)
  • 解决后:
    reshape(-1)

安利时间:

  • 如果本篇博客解决了你的问题,不妨留下你的👍

    让博主知道原来也有小伙伴在同一个坑摔倒了啊hh

  • 欢迎关注大二的博主,在学习C/C++/JAVA/算法/框架的路上作伴

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年5月11日
下一篇 2023年5月11日

相关推荐