change to use torch.device (#3594)

Commit

239 days ago

change to use torch.device (#3594) Signed-off-by: Matrix YAO <matrix.yao@intel.com>