torch torchvision torchmetrics lightning numpy pillow grad-cam albumentations torch_lr_finder tqdm matplotlib pandas seaborn scikit-learn gradio==4.29.0