diff --git a/requirements.txt b/requirements.txt index ad4c783..468b7c2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -torch==1.5.0 +torch==2.2.0 torchvision==0.6.0 pytorch-lightning==0.8.5 \ No newline at end of file