diff --git a/requirements.txt b/requirements.txt index 52e79f5..f1a8b35 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ pillow==5.0.0 scikit-learn==0.19.1 scipy==1.0.0 tensorboard==1.10.0 -tensorflow-gpu==1.10.1 +tensorflow-gpu==1.15.2 torch==0.4.1 torchvision==0.2.1 tqdm==4.23.4