Leo-physique 2022-07-26 21:33 采纳率: 50%
浏览 43
已结题

pytorch复现文章的一个小报错

pytorch 文章复现的小报错
复现了一个backbone，第一次卷积就报错说我输入的图像通道有问题。具体见下图

img

说我的输入是16 * 32 * 510 * 510.

可以我的batch确实是16，但是我的图像是3 * 512 * 512，怎么可能是32 * 510 * 510呢。

于是我打印了一下transform之后的inputs的shape

img

没有发现任何问题。

报错位置代码

就是第一层卷积，扩一下通道

img

肿么了

  • 写回答

3条回答 默认 最新

  • Katsu-S 2022-07-26 22:09
    关注

    输入[16, 3, 512, 512]，经过第一层卷积后变成[16, 32, 510, 510]，这个错误应该说的是后面，看看你第二层卷积的输入通道是不是32

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题？

问题事件

  • 已结题 （查看结题原因） 7月27日
  • 已采纳回答 7月27日
  • 修改了问题 7月26日
  • 创建了问题 7月26日

悬赏问题

  • ¥15 装 pytorch 的时候出了好多问题，遇到这种情况怎么处理？
  • ¥20 IOS游览器某宝手机网页版自动立即购买JavaScript脚本
  • ¥15 手机接入宽带网线，如何释放宽带全部速度
  • ¥30 关于#r语言#的问题：如何对R语言中mfgarch包中构建的garch-midas模型进行样本内长期波动率预测和样本外长期波动率预测
  • ¥15 ETLCloud 处理json多层级问题
  • ¥15 matlab中使用gurobi时报错
  • ¥15 这个主板怎么能扩出一两个sata口
  • ¥15 不是，这到底错哪儿了😭
  • ¥15 2020长安杯与连接网探
  • ¥15 关于#matlab#的问题：在模糊控制器中选出线路信息，在simulink中根据线路信息生成速度时间目标曲线（初速度为20m/s，15秒后减为0的速度时间图像）我想问线路信息是什么