pytorch_lightning==1.5.10
torchmetrics==0.9.1
torchvision==0.12.0
