CUDA out of memory about stdf-pytorch HOT 7 CLOSED

Hi
I guess the program is trying to allocate more gpu memory for testing.
You may turn off the regular testing, or conduct the testing on another gpu.

from stdf-pytorch.

I have commented
# CUDA_VISIBLE_DEVICES=0 python test.py --opt_path option_R3_mfqev2_1G.yml
on the script.sh file but there is no change of behavior.

I just have one GPU on my computer which is used by the OS (linux mint 20.3). With all application closed the system is using an incompressible 500M of GPU memory which I think is causing my problem.

I will try to run in full console mode to free more memory but just in case, can I change some values in the option_R3_mfqev2_1G.yml file to reduce memory consumption ?

from stdf-pytorch.

Nope, tried in full console environment with 0MB of memory used when launching the training script. Same problem.

from stdf-pytorch.

hi, sorry for late response

I mean you may turn off the regular validation in the training process. Check this line:

from stdf-pytorch.

Tank you, it seems to work if i do that.
I think my GPU has no sufficient memory to load 1080p video files.

I tried test the pre-trained models but it fails at the same point.
Also tried test_one_video.py which works fine with the default video (BasketballPass_416x240_500) but when I try to open a 1080p video I got the same error.

from stdf-pytorch.

Glad to hear that!

If you cannot enhance videos with big resolution, you may cut each frame into smaller patches, enhance these patches and finally combine them into one frame.

Or, you may conduct the test process on cpu, which may be 10 times slower than that on gpu.

from stdf-pytorch.

Ok thanks I will try that.

from stdf-pytorch.