diff --git a/tests/metrics/machine_learning/README.md b/tests/metrics/machine_learning/README.md index 1872e9150d..41cc139dcf 100644 --- a/tests/metrics/machine_learning/README.md +++ b/tests/metrics/machine_learning/README.md @@ -1,4 +1,4 @@ -# Kata Containers Tensorflow Metrics +# Kata Containers TensorFlow Metrics Kata Containers provides a series of performance tests using the TensorFlow reference benchmarks (tf_cnn_benchmarks). @@ -30,16 +30,16 @@ Individual tests can be run by hand, for example: $ cd metrics/machine_learning $ ./tensorflow.sh 40 100 ``` -# Kata Containers Tensorflow `MobileNet` Metrics +# Kata Containers TensorFlow `MobileNet` Metrics `MobileNets` are small, low-latency, low-power models parameterized to meet the resource constraints of a variety of use cases. They can be built upon for classification, detection, embeddings and segmentation similar to how other popular large scale models, such as Inception, are used. `MobileNets` can be run efficiently on mobile devices with `Tensorflow` Lite. -Kata Containers provides a test for running `MobileNet V1` inference using Intel-Optimized `Tensorflow`. +Kata Containers provides a test for running `MobileNet V1` inference using Intel-Optimized `TensorFlow`. -## Running the `Tensorflow` `MobileNet` test +## Running the `TensorFlow` `MobileNet` test Individual test can be run by hand, for example: ``` @@ -50,7 +50,7 @@ $ ./tensorflow_mobilenet_benchmark.sh 25 60 # Kata Containers TensorFlow `ResNet50` Metrics `ResNet50` is an image classification model pre-trained on the `ImageNet` dataset. -Kata Containers provides a test for running `ResNet50` inference using Intel optimized +Kata Containers provides a test for running `ResNet50` inference using Intel-Optimized `TensorFlow`. ## Running the `TensorFlow` `ResNet50` test diff --git a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile index 9782d72e4b..a033475ee5 100644 --- a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile +++ b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile @@ -13,9 +13,9 @@ LABEL DOCKERFILE_VERSION="1.0" RUN apt-get update && \ apt-get install -y --no-install-recommends wget nano curl build-essential git && \ apt-get install -y python3.8 python3-pip && \ - pip install intel-tensorflow-avx512==2.8.0 && \ - pip install protobuf==3.20.* && \ - wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_8/resnet50_int8_pretrained_model.pb && \ + pip install --no-cache-dir intel-tensorflow-avx512==2.8.0 && \ + pip install --no-cache-dir protobuf==3.20.* && \ + wget -q https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_8/resnet50_int8_pretrained_model.pb && \ git clone https://github.com/IntelAI/models.git CMD ["/bin/bash"]