diff --git a/requirements.txt b/requirements.txt index 3c6bd02..f93c908 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,5 +2,5 @@ pillow numpy == 1.22.0 setuptools == 78.1.1 tensorboardX == 2.0 -torch == 2.7.0 +torch == 2.8.0 torchvision == 0.5.0 diff --git a/scripts/requirements.txt b/scripts/requirements.txt index a85c02a..1148bd7 100644 --- a/scripts/requirements.txt +++ b/scripts/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.0 +torch>=2.8.0 torchvision tensorboardX pillow