torch>=1.8.1
numpy>=1.19.5
