STN added to Pytorch by Soumith Chintala

[Cadene]

pytorch: Adding Spatial Transformers w/CuDNN support

[fxia22]

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.

[jwyang]

@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!

[jwyang]

@fxia22

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

[jwyang]

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