facebookresearch/supervision-by-registration

the LK operation is very slow, is this normal?

taozhiqi opened this issue · 1 comments

Thanks for your great work. I am trying to use the LK operation in my own work , and i have a question about the Lk operation, i put the LK behind my landmarks model, and i find the LK operation is very slow in the forward propagation, the utilization of gpu is low, is this normal?
Batch_SIZE = 64, INPUT_SIZE =128, single gpu

image

D-X-Y commented

That is normal because we did not optimize its implementation.