search

bubbliiiing / segformer-pytorch

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

不能加载backbone预训练权重 #9

Open worldlyesterday opened 1 year ago

worldlyesterday commented 1 year ago

作者您好，非常感谢您提供的SegFormer-Pytorch实现方法。我在利用您的预训练权重segformer_b2_backbone_weights.pth训练B2网络时发现匹配不上任何键值对： image 请问一下有没有一些解决方法，期待您的解答。

bubbliiiing commented 1 year ago

额pretrained = True

worldlyesterday commented 1 year ago

非常感谢！是我之前没有理解注释的意思：）