pytorch_lightning==1.9.5
torchmetrics==0.9.1
torchvision==0.12.0
