numpy
tensorboardX
tqdm
torch==0.4.1
