diff --git a/requirements.txt b/requirements.txt index 1f4e16f..489a068 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,6 @@ pillow==6.1 scipy==1.1.0 sklearn==0.0 tensorflow==1.12.0 -tensorflow-gpu==1.14.0 +tensorflow-gpu==1.15.4 torch==1.0.1 torchvision==0.2.1