diff --git a/requirements.txt b/requirements.txt index d70bd82..0c392dc 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ matplotlib==3.1.1 numpy==1.17.2 scikit-learn==0.21.3 -torch==1.2.0+cpu +torch==2.2.0 torchvision==0.4.0+cpu progressbar2==3.43.1