diff --git a/runner-requirements.txt b/runner-requirements.txt index 3f40053..f9575d7 100644 --- a/runner-requirements.txt +++ b/runner-requirements.txt @@ -9,7 +9,7 @@ torch>=1.11.0,<2.0.0 torchvision>=0.11.0 pytorch-lightning==1.8.3 -torchmetrics==0.11.0 +torchmetrics==1.2.1 # --------- hydra --------- # hydra-core==1.2.0