torch>=0.4.1
tqdm
tensorboardX
torchvision
