diff --git a/requirements.txt b/requirements.txt index 7f5b22c4..e6fe3b1a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ notebook psutil pytest>=5.0 six -torch==1.4.0 +torch==2.2.0 torchvision tensorboardX>=1.5 tqdm>=4.0