-
Notifications
You must be signed in to change notification settings - Fork 21
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
请问pytorch以及cuda版本对权重影响很大嘛? #27
Comments
pytorch和cuda版本应该不会有显著影响。 |
是的,训练和验证数据使用的都是mfqev2数据集,创建lmdb的代码也没有修改过,就是原封不动跑了几次,都是完全达不到预训练实验效果或者是论文效果。 |
建议试试不用 LMDB 跑。 |
好的,十分感谢!我可以尝试一下。不过可以提供一些思路吗?我个人设想的是直接从yuv序列先裁剪出中心128*128的图片以及前后帧,以多帧图片的形式送入模型中训练吗?(好像这个流程跟LMDB的做法是类似的) |
建议参考一下这个仓库:https://github.com/ryanxingql/powerqe PS. 我没记错的话,这个 stdf-pytorch 仓库评估的应该是 Y-PSNR,因为只对 Y 做了增强。 |
实验室用的是4090,所以测试环境不是10.2,用的11.3,pytorch用的是1.12,但是我在使用原始参数复现的几次最好的效果也只能达到0.61db左右,但是确实使用预训练的模型可以达到0.78db的效果,如果差一点点还能够理解,这个差的是否是有点多了?
The text was updated successfully, but these errors were encountered: