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

RealSR任务的预训练参数模型文件是否放错了? #9

Closed
leomessi17 opened this issue Mar 26, 2024 · 1 comment
Closed

RealSR任务的预训练参数模型文件是否放错了? #9

leomessi17 opened this issue Mar 26, 2024 · 1 comment

Comments

@leomessi17
Copy link

我是从这个链接下载的:https://drive.google.com/file/d/1ZOFzcex2g9_B6Xtf8-qnMx08OnAjKD0M/view?usp=sharing。 下载后发现checkpoint中的keys是这些:
['conv0.weight', 'conv0.bias', 'conv1.weight_orig', 'conv1.weight_u', 'conv1.weight_v', 'conv2.weight_orig', 'conv2.weight_u', 'conv2.weight_v', 'conv3.weight_orig', 'conv3.weight_u', 'conv3.weight_v', 'conv4.weight_orig', 'conv4.weight_u', 'conv4.weight_v', 'conv5.weight_orig', 'conv5.weight_u', 'conv5.weight_v', 'conv6.weight_orig', 'conv6.weight_u', 'conv6.weight_v', 'conv7.weight_orig', 'conv7.weight_u', 'conv7.weight_v', 'conv8.weight_orig', 'conv8.weight_u', 'conv8.weight_v', 'conv9.weight', 'conv9.bias'])
跟MambaIR模型没有关系,是不是放错了?我已经通过日志文件,对齐了模型结构,目前代码打印出来的模型结构和日志中的是一样的,但是加载预训练模型参数时遇到键值对不上的问题。

@csguoh
Copy link
Owner

csguoh commented Mar 27, 2024

Thanks for your kind reminder, we have updated the link of MambaIR-real.pth in the README, now you can try the real-SR model.

@csguoh csguoh closed this as completed Mar 28, 2024
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