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