diff --git a/requirements.txt b/requirements.txt index ba5a442..e270213 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ # --------- pytorch --------- # -pytorch-lightning==1.9.2 +pytorch-lightning==2.0.7 torch==1.13.1 torchmetrics==0.11.1 torchvision==0.14.1