Update torchmetrics from 0.11.4 to 1.3.1

andreeaiana · Mar 18, 2024 · 2ae8f44 · 2ae8f44
1 parent f3ef4d4
commit 2ae8f44
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/environment.yaml b/environment.yaml
@@ -44,7 +44,7 @@ dependencies:
   - sphinx=5.*
   - tokenizers=0.*
   - torchaudio=2.*
-  - torchmetrics=0.*
+  - torchmetrics=1.*
   - tqdm=4.*
   - transformers=4.*
   - wandb=0.*

diff --git a/requirements.txt b/requirements.txt
@@ -26,7 +26,7 @@ tokenizers==0.13.2
 torch==2.0.1
 torchaudio==2.0.2
 torch-geometric==2.3.0
-torchmetrics==0.11.4
+torchmetrics==1.3.1
 torchvision==0.15.2
 tqdm==4.65.0
 transformers==4.36.0

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
         "seaborn==0.12.2",
         "sphinx==5.0.2",
         "tokenizers==0.13.2",
-        "torchmetrics==0.11.4",
+        "torchmetrics==1.3.1",
         "tqdm==4.65.0",
         "transformers==4.36.0",
         "wandb==0.15.3",