fix torch

Change-Id: I0fb6264d5fda2d810675b1f77508159716c06817