pytorch-lightning==1.7.7
wandb==0.15.10
torch==1.12.1
numpy==1.22.4
scikit-image==0.22.0
torchvision==0.13.1
torchmetrics==0.11.4
matplotlib==3.5.1
Pillow==9.0.1
python-dotenv==1.0.1
grad-cam==1.4.8