CUDA out of memory when testing my own datasets
stillbetter opened this issue · 1 comments
stillbetter commented
Hi Dr. Li. Sorry for a third question. When I want to test a self-collected datasets, the test_demo.py give a error feedback that memory is not enough
RuntimeError: CUDA out of memory. Tried to allocate 676.00 MiB (GPU 0; 10.76 GiB total capacity; 1.04 GiB already allocated; 602.56 MiB free; 1.56 GiB reserved in total by PyTorch)
I would guess my video resolution caused this with a 1080*1920 ,however, even the provided test.mp4 get the same error.
I want to know if the all the tests are right in your experiments with any resolution videos?
Thank you!