torch~=1.8.1
torchvision~=0.9.0a0+3f090d0
numpy~=1.20.2
tqdm~=4.60.0
Pillow~=8.2.0
scikit-learn~=0.24.1