Skip to content

Commit adf57f5

Browse files
author
gaosen10
committed
upgrade tensorflow from 1.15.0 to 1.15.2
1 parent 5b5c4de commit adf57f5

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ def run(port, url_queue, result_queue):
282282

283283
def install(model_name):
284284
if model_name == 'face_retrieval':
285-
util.install_package('tensorflow-gpu==1.15.0 mtcnn keras==2.2.4')
285+
util.install_package('tensorflow-gpu==1.15.2 mtcnn keras==2.2.4')
286286
elif model_name == 'image_retrieval':
287287
util.install_package('torchvision torch')
288288
elif model_name == 'text':

src/text/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
Keras==2.2.4
22
keras-bert==0.80.0
33
tensorboard==1.15.0
4-
tensorflow-gpu==1.15.0
4+
tensorflow-gpu==1.15.2

0 commit comments

Comments
 (0)