diff --git a/runner-requirements.txt b/runner-requirements.txt index 3f40053..f2a5809 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.0.9.post0 torchmetrics==0.11.0 # --------- hydra --------- #