torch~=1.7.1
torchvision~=0.8.0a0+ecf4e9c
numpy~=1.19.4
tqdm~=4.58.0
Pillow~=8.1.0
scikit-learn~=0.24.0