diff --git a/requirements.txt b/requirements.txt index cd9e66c..335c132 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # --------- pytorch --------- # torch==2.3.0 -torchvision==0.18.0 +torchvision==0.23.0 lightning torchmetrics>=0.11.4 datasets