torch>=1.1
torchvision
numpy
tqdm
tensorboard>=1.14