diff --git a/requirements.txt b/requirements.txt index ec0e048..d176334 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ gdown==3.7.4 -tensorflow-gpu==1.10.0 +tensorflow-gpu==2.9.3 torch torchvision scikit-image==0.14.2