diff --git a/requirements.txt b/requirements.txt index 092eb54..c45e290 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # --------- pytorch --------- # torch>=1.10.0 torchvision>=0.11.0 -pytorch-lightning==1.6.5 +pytorch-lightning==2.3.2 torchmetrics==0.9.2 # --------- hydra --------- #