diff --git a/requirements.txt b/requirements.txt index 40ebbc1..e43f3ba 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,6 +15,6 @@ google-auth-oauthlib s2cloudless catalyst==22.2.1 segmentation-models-pytorch==0.2.1 -torch==1.7.1 +torch==2.2.0 torchvision==0.8.2 matplotlib==3.6.2