requirements.txt

torch
torchvision
torchaudio
albumentations
lightning
torch-lr-finder
grad-cam
torchsummary
seaborn
pillow
matplotlib
numpy
pandas