numpy==1.20.3
torch==1.8.1
torchvision
