diff --git a/requirements.txt b/requirements.txt index f673bd2..a9920b6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ pandas>=0.25.1 networkx>=2.4 tqdm>=4.43.0 scikit-learn>=0.22.1 -torch==1.4.0 +torch==2.2.0 torchvision==0.5.0 numpy>=1.16.4 scipy>=1.4.1