torch>=1.8.0
numpy>=1.19.2
