torch<2.0.0,>=1.6.0
torchvision<1.0.0,>=0.7.0
