torch>=1.7.1
torchvision>=0.8.2
matplotlib>=3.3.3
seaborn>=0.11.1
scikit-learn>=0.24.0
pyyaml>=5.3.1
pandas>=1.1.5
tqdm>=4.54.1
