diff --git a/requirements.txt b/requirements.txt index e997d8b..d9936e5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ tqdm==4.31.1 -torch==1.0.1.post2 +torch==2.2.0 numpy==1.16.2 tensorboardX==1.7