torch==2.1.2
torchvision
torch-lr-finder
grad-cam
pillow
numpy
pytorch_lightning==1.9.5
lightning-bolts