lliuz / ARFlow

The official PyTorch implementation of the paper "Learning by Analogy: Reliable Supervision from Transformations for Unsupervised Optical Flow Estimation".
MIT License
251 stars 50 forks source link

correlation_cuda module error #20

Open zyl1336110861 opened 3 years ago

zyl1336110861 commented 3 years ago

@lliuz Hello, thanks very much for your sharing! I can run the whole project without any problem. But when I want to measure the running time of your correlation_cuda module by python correlation_native.py, I encounter the cuda error like this: image I don't know how to solve this problem.