pytorch-lightning>=1.5.0
tensorboardX>=2.1
torch>=1.8.0
