diff --git a/docs/installation/env_requirement.rst b/docs/installation/env_requirement.rst index dac154194aef8be952376b769b7d540cf6a01111..82ae473070c0d4126959b775f49a6ecf6c88cdf6 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 764ae9c49e3fd79af99302ee0606253e6409035a..0c564b76ae90af1a6e2318da432e7d31c87602dd 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 39bcd3b88eb86d7024f4ee9d20612d46ac6cf057..9390d5a11af62f2b9cbbe55f2c98ecd4b23ed1a8 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