torch>=1.6.0,<=1.8.0
torchvision>=0.7.0,<=0.9.0