diff --git a/requirements.txt b/requirements.txt index bf1d520..dea64ab 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==1.5.1 +torch==2.2.0 torchvision==0.6.1 albumentations==0.4.5 pytorch_lightning==0.8.1