From 2ae8f44e654edb7dbb4eb9c67eccae84abe8e19b Mon Sep 17 00:00:00 2001 From: andreeaiana Date: Mon, 18 Mar 2024 10:21:23 +0100 Subject: [PATCH] Update torchmetrics from 0.11.4 to 1.3.1 --- environment.yaml | 2 +- requirements.txt | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/environment.yaml b/environment.yaml index 87565b0..1d133f4 100644 --- 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 index b41e60e..dce5679 100644 --- 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 index 2476cf4..5563116 100644 --- 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",