sniklaus/pytorch-liteflownet

is there any way to infer on cpu?

HSILA opened this issue 3 years ago · 1 comments

HSILA commented 3 years ago

is there any way to infer on cpu?

sniklaus commented 3 years ago

The correlation function is written in CUDA: https://github.com/sniklaus/pytorch-liteflownet/blob/master/correlation/correlation.py

Which is why the project does not run on CPUs out of the box, you would have to reimplement the correlation function.

👍1