hack for torch

Change-Id: Id111ae7112122e7a5e44fefa17bf0cd790bde124