diff --git a/src/requirements.txt b/src/requirements.txt index 38d0d35..f9df8a9 100644 --- a/src/requirements.txt +++ b/src/requirements.txt @@ -1,4 +1,4 @@ numpy==1.17.3 scikit-learn==0.21.3 -torch==1.3.0 +torch==2.2.0 torchvision==0.4.1 \ No newline at end of file