diff --git a/requirements.txt b/requirements.txt index 4d444c5..6d0a5ca 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ torch>=1.8.0 torchvision>=0.9.1 pytorch-lightning==1.6.5 -torchmetrics==0.9.2 +torchmetrics==1.1.1 # --------- hydra --------- # hydra-core==1.2.0