search

sniklaus / pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
GNU General Public License v3.0
608 stars 122 forks source link

How to change the cuda version of correlation.py to the python version #62

Closed jiaweiHu-XDU closed 1 year ago

jiaweiHu-XDU commented 1 year ago

How to change the cuda version of correlation.py to the python version, I use PyTorch to calculate according to the same logic, but the final result does not correspond.I would be very grateful if the author wrote a PyTorch version, I really can't write a version that can run on the CPU

sniklaus commented 1 year ago

My apologies for not being able to provide a CPU-only version since I unfortunately don't have bandwidth to do so, maybe you could try reaching out to the authors of PWC-Net through their repository: https://github.com/NVlabs/PWC-Net