pytorch==1.4.0
torchvision==0.4.0
matplotlib