From 6642b17f6e15671878f1040eeee35a7f91112f1e Mon Sep 17 00:00:00 2001 From: liutuo Date: Wed, 19 Dec 2018 10:54:20 +0800 Subject: [PATCH] update tensorflow version to 1.8.0; fix reduce mean convert --- docs/installation/env_requirement.rst | 2 +- docs/installation/manual_setup.rst | 2 +- mace/python/tools/converter_tool/tensorflow_converter.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/installation/env_requirement.rst b/docs/installation/env_requirement.rst index dac15419..82ae4730 100644 --- a/docs/installation/env_requirement.rst +++ b/docs/installation/env_requirement.rst @@ -53,7 +53,7 @@ Optional dependencies - apt-get install android-tools-adb - Required by Android run, >= 1.0.32 * - TensorFlow - - pip install -I tensorflow==1.6.0 + - pip install -I tensorflow==1.8.0 - Required by TensorFlow model * - Docker - `docker installation guide `__ diff --git a/docs/installation/manual_setup.rst b/docs/installation/manual_setup.rst index 764ae9c4..0c564b76 100644 --- a/docs/installation/manual_setup.rst +++ b/docs/installation/manual_setup.rst @@ -65,7 +65,7 @@ Install TensorFlow (Optional) .. code:: sh - pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com tensorflow==1.6.0 + pip install -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com tensorflow==1.8.0 Install Caffe (Optional) diff --git a/mace/python/tools/converter_tool/tensorflow_converter.py b/mace/python/tools/converter_tool/tensorflow_converter.py index 39bcd3b8..9390d5a1 100644 --- a/mace/python/tools/converter_tool/tensorflow_converter.py +++ b/mace/python/tools/converter_tool/tensorflow_converter.py @@ -806,7 +806,7 @@ class TensorflowConverter(base_converter.ConverterInterface): reduce_type_arg = op.arg.add() reduce_type_arg.name = MaceKeyword.mace_reduce_type_str - reduce_type_arg.i = ReduceType.MEAN + reduce_type_arg.i = ReduceType.MEAN.value axis_arg = op.arg.add() axis_arg.name = MaceKeyword.mace_axis_str -- GitLab