torch>=1.2.0
tqdm
scikit-learn
tensorflow
