torch == 1.7.1
transformers == 4.3.2
tensorboardX
scikit-learn
tqdm
