torch>=1.6
numpy==1.19.5
matplotlib==3.3.4
tqdm==4.62.3
PyYAML==6.0
scipy==1.5.4
scikit-learn==0.24.2
