numpy>=1.16.2
torch>=1.0
