search

bubbliiiing / segformer-pytorch

这是一个segformer-pytorch的源码，可以用于训练自己的模型。
MIT License
282 stars 36 forks source link

修改CEloss函数为BCEloss，确保输入输出均为（N, 1）,报错维度不匹配 #6

Closed Buling-Knight closed 2 years ago

Buling-Knight commented 2 years ago

UP主你好，非常感谢你公开的代码，对我帮助很大。 有一个问题想请教一下：我将自己的代码和报错信息截图，请您有空的话指点一下，非常感谢

我尝试修改CEloss为BCEWithLogitsLoss，确保输入尺寸输出尺寸shape相同，仍然报错。 注：我已将解码头输出由（b, 2, h, w）改为（b, 1, h, w）

1 2

Buling-Knight commented 2 years ago

2 这个是补充断点调试的信息。

Buling-Knight commented 2 years ago

weights参数设置的问题，抱歉打扰了昨晚调试调糊涂了

bubbliiiing commented 2 years ago

好的