diff --git a/requirements.txt b/requirements.txt index 092eb54..66dfdc8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ torch>=1.10.0 torchvision>=0.11.0 pytorch-lightning==1.6.5 -torchmetrics==0.9.2 +torchmetrics==1.4.1 # --------- hydra --------- # hydra-core==1.2.0