diff --git a/requirements.txt b/requirements.txt index 5e97adc..e664746 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ numpy>=1.16.0 -torch==1.2.0 +torch==2.2.0 typing>=3.6.6 scikit_learn>=0.20.2 tensorboardX==1.6