torch==1.6.0+cu101
numpy==1.18.5
matplotlib==3.2.2
