Skip to content

Commit acfdcf1

Browse files
authored
Merge pull request #366 from NervanaSystems/jitendra/sync-1.4.1-public-master
Sync up public master for v1.4.1
2 parents ce32c97 + c7d3799 commit acfdcf1

File tree

2 files changed

+15
-5
lines changed

2 files changed

+15
-5
lines changed

Diff for: benchmarks/common/tensorflow/start.sh

+10
Original file line numberDiff line numberDiff line change
@@ -696,6 +696,10 @@ function unet() {
696696
echo "wavenet requires -- checkpoint_name arg to be defined"
697697
exit 1
698698
fi
699+
if [ ${NOINSTALL} != "True" ]; then
700+
# install dependencies
701+
pip3 install --force-reinstall Pillow==5.3.0
702+
fi
699703
if [ ${ACCURACY_ONLY} == "True" ]; then
700704
echo "Accuracy testing is not supported for ${MODEL_NAME}"
701705
exit 1
@@ -797,6 +801,12 @@ function wavenet() {
797801
exit 1
798802
fi
799803

804+
if [ ${NOINSTALL} != "True" ]; then
805+
# install dependencies
806+
apt-get clean && apt-get update -y && \
807+
apt-get install --no-install-recommends --fix-missing libsndfile1 -y
808+
fi
809+
800810
if [[ -z "${sample}" ]]; then
801811
echo "wavenet requires -- sample arg to be defined"
802812
exit 1

Diff for: benchmarks/image_segmentation/tensorflow/unet/README.md

+5-5
Original file line numberDiff line numberDiff line change
@@ -20,17 +20,17 @@ modes/precisions:
2020
```
2121

2222
3. Clone the [tf_unet](https://github.com/jakeret/tf_unet) repository,
23-
and then get [PR #202](https://github.com/jakeret/tf_unet/pull/202)
23+
and then get [PR #276](https://github.com/jakeret/tf_unet/pull/276)
2424
to get cpu optimizations:
2525

2626
```
2727
$ git clone [email protected]:jakeret/tf_unet.git
2828
2929
$ cd tf_unet/
3030
31-
$ git fetch origin pull/202/head:cpu_optimized
31+
$ git fetch origin pull/276/head:cpu_optimized
3232
From github.com:jakeret/tf_unet
33-
* [new ref] refs/pull/202/head -> cpu_optimized
33+
* [new ref] refs/pull/276/head -> cpu_optimized
3434
3535
$ git checkout cpu_optimized
3636
Switched to branch 'cpu_optimized'
@@ -60,7 +60,7 @@ modes/precisions:
6060
--docker-image gcr.io/deeplearning-platform-release/tf-cpu.1-14 \
6161
--checkpoint /home/<user>/unet_trained \
6262
--model-source-dir /home/<user>/tf_unet \
63-
-- checkpoint_name=model.cpkt
63+
-- checkpoint_name=model.ckpt
6464
```
6565

6666
Note that the `--verbose` or `--output-dir` flag can be added to the above
@@ -75,4 +75,4 @@ modes/precisions:
7575
Total samples/sec: 905.5344 samples/s
7676
Ran inference with batch size 1
7777
Log location outside container: {--output-dir value}/benchmark_unet_inference_fp32_20190201_205601.log
78-
```
78+
```

0 commit comments

Comments
 (0)