pytorch

fb68b0a5 - modify check to make it consistent (not sure if this works??)