torch==1.8.1
scipy==1.6.1
numpy
matplotlib
tqdm