jwyang / lr-gan.pytorch

Pytorch code for our ICLR 2017 paper "Layered-Recursive GAN for image generation"
151 stars 32 forks source link

Pytorch stnm #15

Open kwon-young opened 5 years ago

kwon-young commented 5 years ago

Implementation of the STNM module in pytorch. also use the native torch.nn.functional.affine_grid to generate the grid.