pytorch-lightning==1.6.5
torchvision==0.12.0
torch==1.11.0
torchmetrics==0.7.3
