diff --git a/requirements/base.txt b/requirements/base.txt index d699c72e91..c8d6bb13ac 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -4,7 +4,7 @@ packaging <23.0 setuptools <=59.5.0 # Prevent install bug with tensorboard numpy <1.24 # strict - freeze for using np.long torch >1.7.0 -torchmetrics >0.7.0, <0.11.0 # strict +torchmetrics >0.7.0, <0.12.0 # strict pytorch-lightning >1.6.0, <1.9.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 6d88a049f4..305816c7a2 100644 --- a/requirements/datatype_image_extras.txt +++ b/requirements/datatype_image_extras.txt @@ -13,7 +13,7 @@ fairscale pytorch-lightning <1.5.0 # 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