torch>=1.1.0
transformers>=4.0.0
scikit-learn
tensorboardX
