Pytorch.transforms

A small fix has been deployed, so it is now possible to update pytorch, I faced some issues with the latest available version (0.3.1), but 0.3.0 (cuda 8.0) works well for me.