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.

Sign up for GitHub

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

Jump to bottom

关于BD退化模型 #50

Open
qizhou000 opened this issue Jul 8, 2021 · 5 comments
Open

关于BD退化模型 #50

qizhou000 opened this issue Jul 8, 2021 · 5 comments

Comments

@qizhou000
Copy link

您好！请问BDx3的模型的训练权重能提供一下吗？我需要用于定性比较
@guihaik
Copy link

guihaik commented Aug 5, 2021

您好！请问BDx3的模型的训练权重能提供一下吗？我需要用于定性比较

您好，我用提供的pretrain_model中的SAN_BI4X.pt在Set5上的psnr值为18.937，请问您有类似的问题吗？

@qizhou000
Copy link
Author

您好！请问BDx3的模型的训练权重能提供一下吗？我需要用于定性比较

您好，我用提供的pretrain_model中的SAN_BI4X.pt在Set5上的psnr值为18.937，请问您有类似的问题吗？

这倒是没有，我是正常的，我猜你的RGB范围和SAN的不一样，看一下，[0, 1]和[0, 255]

@guihaik
Copy link

guihaik commented Aug 5, 2021

您好！请问BDx3的模型的训练权重能提供一下吗？我需要用于定性比较

您好，我用提供的pretrain_model中的SAN_BI4X.pt在Set5上的psnr值为18.937，请问您有类似的问题吗？

这倒是没有，我是正常的，我猜你的RGB范围和SAN的不一样，看一下，[0, 1]和[0, 255]

谢谢您，确实是RGB的问题。

@EchoXu98
Copy link

求助！请问兄弟们是用几块GPUs训练的啊？我用一块1080ti，12G显存的卡之前一直在MPNCOV.py大约第24行那里报CUDA out of memory,我参考别人给的代码对24行那里做了修改，同时只有将训练时的batch_size修改为1时，在训练时才不会报错。但是在测试时又在MPNCOV.py文件下报了CUDA out of memory错误。

@qizhou000
Copy link
Author

求助！请问兄弟们是用几块GPUs训练的啊？我用一块1080ti，12G显存的卡之前一直在MPNCOV.py大约第24行那里报CUDA out of memory,我参考别人给的代码对24行那里做了修改，同时只有将训练时的batch_size修改为1时，在训练时才不会报错。但是在测试时又在MPNCOV.py文件下报了CUDA out of memory错误。

测试时用chop，它显存很大

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@guihaik @qizhou000 @EchoXu98