torch>=1.5.0
torchvision>=0.6.1
numpy>=1.16.4
