torch>=1.3.1
numpy>=1.19.5
