numpy
torch>=1.1
tensorboard
tqdm
