diff --git a/requirements.txt b/requirements.txt index ba5a442..d5f9b5a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # --------- pytorch --------- # pytorch-lightning==1.9.2 torch==1.13.1 -torchmetrics==0.11.1 +torchmetrics==1.0.0 torchvision==0.14.1 # --------- hydra --------- #