Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

代码提供的预训练权重和Vmamba中提供权重该使用哪一个更好? #72

Open
lwtgithublwt opened this issue Aug 10, 2024 · 1 comment

Comments

@lwtgithublwt
Copy link

因为看到文章说采用的vmamba中的预训练权重,但代码也提供了自己的预训练权重,这和Vmamba中提供的有什么区别,具体应该使用哪一个,还是都可以使用?
此外,由于修改了部分代码加载时显示如,Not loaded keys: ['layers_up.3.blocks.0.norm.weight.....这是否影响网络的训练与结果?非常感谢您的回答!

@JCruan519
Copy link
Owner

建议直接使用我提供的预训练权重,VMAMBA的权重更新过一版,不再适配VM-UNet了

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants