tensorboard==2.1.1
torch==1.5.0
torchvision==0.6.0