Try newer torchmetrics

microsoft · Feb 22, 2025 · 801c0ba · 801c0ba
1 parent c4b5d17
commit 801c0ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/min-reqs.old b/requirements/min-reqs.old
@@ -18,7 +18,7 @@ segmentation-models-pytorch==0.3.3
 shapely==1.8.0
 timm==0.9.2
 torch==1.13.0
-torchmetrics==0.10.0
+torchmetrics==1.6.1
 torchvision==0.14.0
 typing-extensions==4.5.0