use torch config

Change-Id: Ibedf7c7c43fcfa24fa51a9114a74576a4ad2c04a