pytorch

e1c5bc99 - removing unneeded device variable