diff --git a/requirements.txt b/requirements.txt index 6bf6f86..c6ee812 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ numpy==1.13.3 -torch==1.4.0 +torch==2.2.0 torchvision==0.5.0