numpy>=1.17.1
matplotlib>=3.0.3
visdom>=0.1.8.8
torchinfo>=1.5.2
torchviz>=0.0.2
torch>=1.8.1
torchvision>=0.9.1
