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

a question of cuda out of memory in test one video #20

Closed

LHL731 opened this issue Jul 11, 2022 · 1 comment

LHL731 commented Jul 11, 2022

您好，当我使用一个视频进行测试时，我用416240与832480的yuv序列进行测试的时候没有问题，但使用更高分辨率的yuv序列进行测试的时候就会爆显存，我有看到训练的时候会裁剪成128*128的，但是对一个视频进行测试的时候为了更好的显示增强结果就没有进行图片裁剪，想问问您是怎么解决这个问题的呢

Owner

ryanxingql commented Jul 11, 2022

您好，目前框架确实存在这一问题，例如BasicVSR++也一样

我建议通过切块的方法做。代码和示例可以参考这里：
https://gist.github.com/ryanxingql/e126eeb0d2e156bcfe28bc532143d82e

块可以在显存允许的情况下尽量大一些，减少对性能的影响。

ryanxingql closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment