diff --git a/requirements.txt b/requirements.txt index d4661a3..544e2d3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ pytorch-lightning==1.6.0 torch==1.11 timm==0.5.4 torchvision==0.12.0 -scikit-learn==1.0.2 +scikit-learn==1.5.0 opencv-python==4.5.3.56 numpy==1.21.2 albumentations==1.1.0 \ No newline at end of file