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