search

daitao / SAN

Second-order Attention Network for Single Image Super-resolution (CVPR-2019)
550 stars 106 forks source link

关于BD退化模型 #50

Open qizhou000 opened 3 years ago

qizhou000 commented 3 years ago

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

guihaik commented 2 years ago

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

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

qizhou000 commented 2 years ago

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

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

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

guihaik commented 2 years ago

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

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

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

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

EchoXu98 commented 1 year ago

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

qizhou000 commented 1 year ago

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

测试时用chop，它显存很大