Submit failed: No module named 'torch'

[yitian_chen]

Thanks a lot ~it works