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