diff --git a/requirements.txt b/requirements.txt index 4b99c7a97..7783b23a8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # --------- pytorch --------- # torch>=1.10.0 torchvision>=0.11.0 -pytorch-lightning==1.8.3 +pytorch-lightning==1.9.1 torchmetrics==0.11.0 # --------- hydra --------- #