Open kwon-young opened 5 years ago
Implementation of the STNM module in pytorch. also use the native torch.nn.functional.affine_grid to generate the grid.
Implementation of the STNM module in pytorch. also use the native torch.nn.functional.affine_grid to generate the grid.