torch
torchvision==0.20.0
