torch>=1.4.0
numpy>=1.18.1
