torch
torch-lr-finder
torchvision
pillow
gradio
grad-cam
numpy
pytorch-lightning
albumentations