pytorch_lightning<1.7.0,>=1.5.0
torchmetrics<=0.7.3,>=0.7.0
scipy
scikit-learn
