pytorch-lightning>=1.5.0,<=2.1.2
tensorboardX>=2.1
torch>=1.8.0
