matplotlib>=3.2.1
numpy>=1.18.2
torch>=1.4.0
torchvision>=0.5.0
tqdm>=4.45.0
scikit-learn>=0.24.0
seaborn>=0.11.1