segformer-pytorch icon indicating copy to clipboard operation
segformer-pytorch copied to clipboard
verified publisher icon bubbliiiing

如何添加B5 backbone权重

Open BigBigOranges opened this issue 2 years ago • 3 comments

主干网络设置为B5，,Pretrain设置为True，会自动从Github上下载你的backbone权重。但是目前仓库只上传了b0,b1,b2，我该如何添加B5 backbone权重进行预训练呢？

BigBigOranges avatar Nov 07 '23 03:11 BigBigOranges

作者给了百度云连接下载好 他自动就读进去了

uoe-haoyu avatar Nov 13 '23 13:11 uoe-haoyu

似乎b2_voc的权重 与b5的权重不一致，当我是用b5测试推理predict时，权重会载入错误，模型主干网结构我也修改为b5了？

hua2014 avatar Dec 28 '23 07:12 hua2014

原来voc的是 作者自己预训练的模型 赞啊

hua2014 avatar Dec 28 '23 07:12 hua2014