torch<1.5.0,>=1.0.0
torchvision>=0.4.0
numpy
click
matplotlib
torchviz
