diff --git a/requirements.txt b/requirements.txt index e04e01b..86dd3c0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ numpy==1.16.4 str2bool==1.1 -torch==1.2.0 +torch==2.2.0 torchvision==0.4.0