torch>=1.1.0
tqdm
sklearn
tensorflow
