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
