STN added to Pytorch by Soumith Chintala

Question

STN added to Pytorch by Soumith Chintala

Cadene opened this issue 8 years ago · 4 comments

pytorch: Adding Spatial Transformers w/CuDNN support

Answer 1 · 2017-07-13T16:33:10.000Z

Thanks @Cadene, I am aware of it. I am going to fix the existing issues for this repo when I have a chance for current users. But given STN is added to pytorch with higher performance, it's unlikely I will develop this repo further.

Answer 2 · 2017-09-16T22:50:22.000Z

@fxia22 ,

Hi, Fei, thanks a lot for sharing this code. I compared with Pytorch official STN and your implementation, and found the gradients are different when the number of channel of feature map > 1. I am wondering whether you also found this, or it's just me having something wrong?

thanks!

Answer 3 · 2017-09-16T22:51:40.000Z

@fxia22

One more point, when channel > 1, the gradients of the first channel are the same, the other channels are different.

Answer 4 · 2017-09-17T06:13:57.000Z

I found the problem, it is not about the cuda code, it on my own code. never mind.