From 462e1ae7f91fe34492206f20a05a38131e96aea9 Mon Sep 17 00:00:00 2001 From: Sergey Kanzhelev Date: Wed, 20 Apr 2022 00:14:21 +0000 Subject: [PATCH] fix the image for node performance tests - model expected tensorflow version <1.9 Change-Id: I116cc38eac20f7cdafb975c88ee9a0e7ec667861 --- test/images/node-perf/tf-wide-deep/Dockerfile | 6 +++--- test/images/node-perf/tf-wide-deep/README.md | 2 +- test/images/node-perf/tf-wide-deep/VERSION | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/test/images/node-perf/tf-wide-deep/Dockerfile b/test/images/node-perf/tf-wide-deep/Dockerfile index 37eac1e6dd1..e0e8bc617c1 100644 --- a/test/images/node-perf/tf-wide-deep/Dockerfile +++ b/test/images/node-perf/tf-wide-deep/Dockerfile @@ -21,10 +21,10 @@ RUN apt-get update && apt-get install -y wget time RUN case $(uname -m) in \ aarch64) \ - pip install tensorflow-aarch64; \ + pip install tensorflow-aarch64==1.2; \ ;; \ *) \ - pip install tensorflow; \ + pip install tensorflow==1.9.0; \ ;; \ esac @@ -35,4 +35,4 @@ RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \ WORKDIR $HOME/models-1.9.0/official/wide_deep ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0 -ENTRYPOINT python ./wide_deep.py +ENTRYPOINT python ./data_download.py && python ./wide_deep.py diff --git a/test/images/node-perf/tf-wide-deep/README.md b/test/images/node-perf/tf-wide-deep/README.md index d69ffac20d2..4df11c85209 100644 --- a/test/images/node-perf/tf-wide-deep/README.md +++ b/test/images/node-perf/tf-wide-deep/README.md @@ -2,7 +2,7 @@ The container image described here predicts the income using the census income dataset in Tensorflow. For more information, see -[https://github.com/tensorflow/models/tree/master/official/r1/wide_deep](https://github.com/tensorflow/models/tree/master/official/r1/wide_deep). +[https://github.com/tensorflow/models/tree/v2.0/official/r1/wide_deep](https://github.com/tensorflow/models/tree/v2.0/official/r1/wide_deep). This image is used as a workload in node performance testing. ## How to release: diff --git a/test/images/node-perf/tf-wide-deep/VERSION b/test/images/node-perf/tf-wide-deep/VERSION index 9459d4ba2a0..5625e59da88 100644 --- a/test/images/node-perf/tf-wide-deep/VERSION +++ b/test/images/node-perf/tf-wide-deep/VERSION @@ -1 +1 @@ -1.1 +1.2