Add detach to the input of Localization Net

aicaffeinelife / Pytorch-STN

Spatial Transformer Networks in Pytorch.

MIT License

155 stars 33 forks source link

Closed Akimoto-Cris closed 4 years ago

Akimoto-Cris commented 5 years ago

the input feature map doesn't require gradients from localization net

aicaffeinelife commented 4 years ago

I didn't see this until now. You're right, the localization net should not propagate gradients back to the input node. I'll merge this PR. Thanks!