Question

模型训练中添加预训练模型报错，请帮忙看一下错误原因，图一为模型训练参数设置，图二为预训练模型的日志

Answer 1

您好，日志倒数后三行有提示报错信息，意思是预训练权重文件（checkpoint）中，该参数的形状是 [64, 3, 7, 7]，与Segformer

模型中，该参数的形状是 [64, 4, 7, 7]不匹配。您可以修改下您的预训练模型，或者换一个匹配的模型。

希望我的回答能帮助到您。

Comments

您好 请问预训练模型是需要修改哪些数据呢，或者匹配的模型是不使用segformer的模型吗，还是要更改模型文件的一些基础配置呢

---

您好，就是说数据和模型没有匹配，segformer需要通道数为3的数据，您可以检查一下您的数据是否通道数为3，可以在idesktopx点开属性，里面有位深数值

---

好的 谢谢啦