I am using the demo.py code on a relatively large image size of 1408x1408. However, it is raising a CUDA error: illegal memory access was encountered.
This seems to be happening after the line flow_low, flow_up = model(image1, image2, iters=50, test_mode=True)
After this line, trying to access flow_up or image1 leads to the above error. I suspect this might be due to cuda memory issues? When I use a smaller image this error does not occur. Any idea how I can resolve this? Thank you!
I am using the demo.py code on a relatively large image size of 1408x1408. However, it is raising a CUDA error: illegal memory access was encountered.
This seems to be happening after the line
flow_low, flow_up = model(image1, image2, iters=50, test_mode=True)After this line, trying to access flow_up or image1 leads to the above error. I suspect this might be due to cuda memory issues? When I use a smaller image this error does not occur. Any idea how I can resolve this? Thank you!