tqdm
numpy
torch>=0.4.0