pytorch-lightning==1.6.5
omegaconf
torchvision