search

bubbliiiing / segformer-pytorch

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

如何添加B5 backbone权重 #23

Open BigBigOranges opened 12 months ago

BigBigOranges commented 12 months ago

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

uoe-haoyu commented 11 months ago

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

hua2014 commented 10 months ago

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

hua2014 commented 10 months ago

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