pytorch-lightning >= 1.0.0rc2
torch >= 1.3.0
torchvision >= 0.6.0
transformers>=4.4.0