weight of size[1408,352,1,1]expected input[8,384,9,9]to have 352channels,but got 384 instead

更换模型时,经常出现这个问题,大概率是通道不匹配,首先确认自己的图片是黑白还是彩色,其次看通道,首先找到根据报错断点位置,发现断点之后,加一行print(x.shape)查看当前的尺寸,按照这个尺寸进行调整即可。下面是我碰到的问题该错过程,我是在efficientnet中更换模型b2时产生错误,但是参数都来自官网,我不知道为什么报错,不过可以解决这个问题。

具体参数可以看这位兄弟的博客(17条消息) EfficientNet B0-B7 网络参数_繁华落尽,寻一世真情的博客-CSDN博客_efficientnet参数量下面是我的详细改错过程

 

报错在这里

找到这里

 

 

打印出来的尺寸,竟然是这样的,按照这个更改

改成这样

如此图所示,程序开始正常运行了 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值