torch>=1.1
scikit-learn
numpy
pandas
tqdm
lmdb