Update torchvision requirement from <0.21,>=0.16.2 to >=0.16.2,<0.22

Update torchvision requirement from <0.21,>=0.16.2 to >=0.16.2,<0.22 #482