uplift torchmetrics, torchvision deps

Lightning-Universe · May 5, 2024 · fbc8f2f · fbc8f2f
1 parent e4c3dd6
commit fbc8f2f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements/base.txt b/requirements/base.txt
@@ -1,6 +1,6 @@
 numpy <1.26.0
 lightning >=2.0.0  # strict
-torchmetrics >=0.10.0, <0.12.0
+torchmetrics >=0.7.0, <1.3.0
 lightning-utilities >0.3.1  # this is needed for PL 1.7
-torchvision >=0.10.0  # todo: move to topic related extras
+torchvision >=0.15.0, <0.19.0  # todo: move to topic related extras
 tensorboard >=2.9.1, <2.14.0  # for `TensorBoardLogger`
diff --git a/requirements/models.txt b/requirements/models.txt
@@ -1,4 +1,4 @@
-torchvision >=0.10.0
+torchvision >=0.15.0, <0.19.0
 scikit-learn >=1.0.2
 Pillow >9.0.0
 gym[atari] >=0.17.2, <0.22.0  # strict