diff --git a/requirements.txt b/requirements.txt index b33536b..7caaecd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==2.1.* +torch==2.2.* torchvision==0.16.* mmcv==2.1.* # Update CUDA Version if necessary.