torch==2.0.1
torchvision==0.15.2
lightning==2.0.6
albumentations==1.3.1
torch-lr-finder
torchinfo
grad-cam
torchmetrics==0.8.0
pandas
datasets==2.14.4
tokenizers==0.13.3