torch~=1.8.0
torchvision~=0.9.0a0+3f090d0
numpy~=1.20.1
tqdm~=4.59.0
Pillow~=8.1.0
scikit-learn~=0.24.0