pytorch

4306ca29 - Minor cleanup, remove useless code added before