torch>=1.2.0
numpy>=1.16.4
