diff --git a/requirements.txt b/requirements.txt old mode 100755 new mode 100644 index e6fdc67..902b0dc --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # --------- pytorch --------- # torch>=2.0.0 torchvision>=0.15.0 -lightning>=2.0.0 +lightning>=2.6.4 torchmetrics>=0.11.4 # --------- hydra --------- #