diff --git a/requirements.txt b/requirements.txt index 7e1950a0..cf806392 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ requests scikit-image scipy tb-nightly -torch>=1.7.1 +torch>=2.8.0 torchvision tqdm yapf