diff --git a/requirements/base.txt b/requirements/base.txt index d2112e31b1..e2d3db7dcf 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -4,7 +4,7 @@ packaging <24.0 setuptools <=59.5.0 # Prevent install bug with tensorboard numpy <1.24 # strict - freeze for using np.long torch >1.7.0, <=2.0.1 -torchmetrics >0.7.0, <0.11.0 # strict +torchmetrics >0.7.0, <0.12.0 # strict pytorch-lightning >1.8.0, <2.0.0 # strict pyDeprecate >0.1.0 pandas >1.1.0, <=1.5.2 diff --git a/requirements/datatype_image_extras.txt b/requirements/datatype_image_extras.txt index b70467af3e..c6060fc59e 100644 --- a/requirements/datatype_image_extras.txt +++ b/requirements/datatype_image_extras.txt @@ -10,7 +10,7 @@ fastface <=0.1.3 # dead fairscale # pinned PL so we force a compatible TM version -torchmetrics<0.8.0 +torchmetrics<0.12.0 # effdet had an issue with PL 1.12, and icevision doesn't support effdet's latest version yet (0.3.0) torch <1.12