diff --git a/requirements.txt b/requirements.txt index 092eb54..38c9481 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==0.11.0 # --------- hydra --------- # hydra-core==1.2.0