matplotlib==3.2.1
numpy==1.18.4
scikit-learn==0.23.2
scipy==1.5.2
torch==1.7.1
torchvision==0.8.2
tqdm==4.46.0