torch>=1.6.0
numpy>=1.19.1
