Fix torch
Merge pull request #1 from Abhinav1997/master

Fix torch