search

fxia22 / stn.pytorch

pytorch version of spatial transformer networks
Other
587 stars 87 forks source link

Added full cuda support #3

Closed jj0mst closed 7 years ago

jj0mst commented 7 years ago

With this one can directly use stn on cuda tensors