search

fxia22 / stn.pytorch

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

Add cuda support to AffineGridGenV2 module #8

Closed aarcosg closed 7 years ago

fxia22 commented 7 years ago

Thanks