search

lizhihao6 / Forward-Warp

An optical flow forward warp's lib with backpropagation using pytorch.
MIT License
109 stars 17 forks source link

About total_step-1 #7

Closed tkkcc closed 3 years ago

tkkcc commented 3 years ago

Hi, I'm confused about this -1. It looks like a mistake to me. Can you explain the purpose? https://github.com/lizhihao6/Forward-Warp/blob/27fe46cd766d960e10d07b3a3156816a8e63ed8e/Forward_Warp/cuda/forward_warp_cuda_kernel.cu#L35

lizhihao6 commented 3 years ago

It seems like a bug. Have you tried with total_step instead of total_step-1?

tkkcc commented 3 years ago

Yes, I use total_step. It seems ok.

lizhihao6 commented 3 years ago

Thx, I have fixed that.