Skip to content

Commit ea1b2e7

Browse files
ashahbajitendra42
authored andcommitted
Fix Pillow version used by unet model (#359)
1 parent 53e25d0 commit ea1b2e7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

benchmarks/common/tensorflow/start.sh

Lines changed: 4 additions & 0 deletions
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

0 commit comments

Comments
 (0)