diff --git a/requirements.txt b/requirements.txt index a24e106..81f01f5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ torch>=1.10.0 torchvision>=0.11.0 pytorch-lightning==2.0.0 -torchmetrics==0.11.0 +torchmetrics==1.2.0 # --------- hydra --------- # hydra-core==1.3.2