torch==1.12.1+cu116
omegaconf==2.2.3
scikit-learn==1.1.2
scipy==1.9.3
pytorch-lightning==1.8.6
