Updated torch related things

Change-Id: If8a5b95476f00b30f1385522966fa84921ce553e
Signed-off-by: Josue Rivera <prbassplayer@gmail.com>