diff --git a/requirements.txt b/requirements.txt index 775922b..3b3d0f4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,5 +11,5 @@ scikit-learn>=0.19.0 scipy>=1.0.0 scikit-image>=0.16.1 seaborn>=0.9.0 -torch==1.2.0 +torch==2.2.0 torchvision==0.2.2.post3