pytorch-lightning==1.8.6
torch==1.12.1
scikit-learn==1.1.2
omegaconf==2.2.3
matplotlib==3.6.3
matplotlib-inline==0.1.6
torchvision==0.13.1
