pytorch version

[skrish13]

Thanks for the awesome repo!

torch.nn.functional.grid_sample doesn't have align_corners in torch < 1.3.0, so you might wanna change the requirements or change the code :)