Merge pull request #366 from NervanaSystems/jitendra/sync-1.4.1-publi…

…c-master Sync up public master for v1.4.1
intel · Jul 10, 2019 · acfdcf1 · acfdcf1
2 parents ce32c97 + c7d3799
commit acfdcf1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/benchmarks/common/tensorflow/start.sh b/benchmarks/common/tensorflow/start.sh
@@ -696,6 +696,10 @@ function unet() {
       echo "wavenet requires -- checkpoint_name arg to be defined"
       exit 1
     fi
+    if [ ${NOINSTALL} != "True" ]; then
+      # install dependencies
+      pip3 install --force-reinstall Pillow==5.3.0
+    fi
     if [ ${ACCURACY_ONLY} == "True" ]; then
       echo "Accuracy testing is not supported for ${MODEL_NAME}"
       exit 1
@@ -797,6 +801,12 @@ function wavenet() {
       exit 1
     fi
 
+    if [ ${NOINSTALL} != "True" ]; then
+      # install dependencies
+      apt-get clean && apt-get update -y && \
+      apt-get install --no-install-recommends --fix-missing libsndfile1 -y
+    fi
+
     if [[ -z "${sample}" ]]; then
       echo "wavenet requires -- sample arg to be defined"
       exit 1

diff --git a/benchmarks/image_segmentation/tensorflow/unet/README.md b/benchmarks/image_segmentation/tensorflow/unet/README.md
@@ -20,17 +20,17 @@ modes/precisions:
    ```
 
 3. Clone the [tf_unet](https://github.com/jakeret/tf_unet) repository,
-   and then get [PR #202](https://github.com/jakeret/tf_unet/pull/202)
+   and then get [PR #276](https://github.com/jakeret/tf_unet/pull/276)
    to get cpu optimizations:
 
    ```
    $ git clone git@github.com:jakeret/tf_unet.git
 
    $ cd tf_unet/
 
-   $ git fetch origin pull/202/head:cpu_optimized
+   $ git fetch origin pull/276/head:cpu_optimized
    From github.com:jakeret/tf_unet
-    * [new ref]         refs/pull/202/head -> cpu_optimized
+    * [new ref]         refs/pull/276/head -> cpu_optimized
 
    $ git checkout cpu_optimized
    Switched to branch 'cpu_optimized'
@@ -60,7 +60,7 @@ modes/precisions:
        --docker-image gcr.io/deeplearning-platform-release/tf-cpu.1-14 \
        --checkpoint /home/<user>/unet_trained \
        --model-source-dir /home/<user>/tf_unet \
-       -- checkpoint_name=model.cpkt
+       -- checkpoint_name=model.ckpt
    ```
 
    Note that the `--verbose` or `--output-dir` flag can be added to the above
@@ -75,4 +75,4 @@ modes/precisions:
    Total samples/sec: 905.5344 samples/s
    Ran inference with batch size 1
    Log location outside container: {--output-dir value}/benchmark_unet_inference_fp32_20190201_205601.log
-   ```
+   ```