Performance issue about pytorch-pwc

[sniklaus]

What is the input resolution for the table that you quoted? And which graphics card are you using?

[donnydonny123]

Hi,
In the original paper's conclusion, 1024*436 images is used for benchmark.
The command I am using is the same as provided in https://github.com/sniklaus/pytorch-pwc#usage ,
Which is python run.py --model default --first ./images/first.png --second ./images/second.png --out ./out.flo
GPU is GTX 1080 ti.
Thank you for your reply.

[sniklaus]

Interesting find, thank you for bringing this up! Calling moduleNetwork in a for loop using the provided images clocks in at around 130 ms per estimate on my 1080 Ti, still much too slow. Note to everyone reading this and who might not be aware of the asynchronous computation nature of GPU computing, making sure to use torch.cuda.synchronize() to get the correct timing. Have you tried benchmarking the official release? If so, what is the timing you get? Thanks!

[sniklaus]

Closing the issue due to inactivity, I would still love to hear more about this though!