diff --git a/requirements.txt b/requirements.txt index 650fc46..f217ad6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ numpy==1.18.4 python==3.6.10 tensorboard==2.0.0 -torch==1.5.0 +torch==2.2.0 torchvision==0.6.0 matplotlib==3.2.1 scikit-learn==0.23.1