CUDA error: an illegal memory access was encountered

[Davidyao99]

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!