pytorch

ed5a3470 - fix some formatting issues and add paramter