这个错误提示是Conv2d的输入通道数(C_in)不等于卷积核的通道数(也是C_in)。可能是因为模型的输入通道数不是3,而是224。

建议检查模型的输入通道数是否正确,并根据需要进行调整。另外,也可以检查输入数据的通道数是否正确。

MindSpore: 解决Conv2d输入通道数错误

原文地址: https://www.cveoy.top/t/topic/mUUg 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录