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