numpy
pyaml
matplotlib
torch==1.3.1
torchvision==0.4.2