diff --git a/requirements.txt b/requirements.txt index f748b5b..3c90f44 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==1.0.0 +torch==2.2.0 numpy>=1.16.0 tqdm>=4.30.0 tensorboardX>=1.6 \ No newline at end of file