numpy
torch
visdom>=0.1.8.8
tensorboardX==1.4
