diff --git a/requirements.txt b/requirements.txt index cd9e66c..477d851 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ # --------- pytorch --------- # -torch==2.3.0 +torch==2.8.0 torchvision==0.18.0 lightning torchmetrics>=0.11.4