torch>=1.3.1
transformers==4.3.2
tqdm
scipy
datasets
sklearn