stuff related to torch version change