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