diff --git a/requirements.txt b/requirements.txt index 94cd808..623f0cb 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # --------- pytorch --------- # torch>=2.0.0 -torchvision>=0.15.0 +torchvision>=0.27.0 lightning>=2.0.0 torchmetrics>=0.11.4