torch~=1.7.1 torchvision~=0.8.0a0+ecf4e9c numpy~=1.19.4