diff --git a/requirements/base.txt b/requirements/base.txt index e460970415..b3e826290f 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.25 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.2.0 pandas >1.1.0, <=1.5.2 diff --git a/requirements/datatype_image_extras.txt b/requirements/datatype_image_extras.txt index fbc686064c..1934a6ccc9 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 diff --git a/requirements/testing_serve.txt b/requirements/testing_serve.txt index 99e4817627..cb127a1379 100644 --- a/requirements/testing_serve.txt +++ b/requirements/testing_serve.txt @@ -10,4 +10,4 @@ sahi ==0.8.19 torch ==1.10.2 torchaudio ==0.10.2 torchvision ==0.11.3 -torchmetrics ==0.10.3 +torchmetrics ==0.11.4