Upgrade dependency to torchmetrics == 1.0.1 (#205)

IntelLabs · Jul 20, 2023 · 74600c4 · 74600c4
1 parent 2cfc963
commit 74600c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ dependencies = [
   "torchvision ~= 0.15.2",
   "lightning[extra] ~= 2.0.5", # Full functionality including TensorboardX.
   "pydantic == 1.10.11", # https://github.com/Lightning-AI/lightning/pull/18022/files
-  "torchmetrics == 1.0.0",
+  "torchmetrics == 1.0.1",
   "numpy == 1.23.5", # https://github.com/pytorch/pytorch/issues/91516
 
   # --------- hydra --------- #