From 7bc387068d4157b25a8a8de943b434396780f4d1 Mon Sep 17 00:00:00 2001 From: liyin Date: Wed, 21 Aug 2019 15:07:18 +0800 Subject: [PATCH] Fix case output type --- tools/python/transform/tensorflow_converter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/python/transform/tensorflow_converter.py b/tools/python/transform/tensorflow_converter.py index 7d4057fc..7f940c1d 100644 --- a/tools/python/transform/tensorflow_converter.py +++ b/tools/python/transform/tensorflow_converter.py @@ -1007,11 +1007,11 @@ class TensorflowConverter(base_converter.ConverterInterface): if dtype == tf.int32: op.output_type.extend([mace_pb2.DT_INT32]) elif dtype == tf.float32: - op.output_type.extend([self._option.data_type]) + op.output_type.extend([mace_pb2.DT_FLOAT]) else: mace_check(False, "data type %s not supported" % dtype) except ValueError: - op.output_type.extend([self._option.data_type]) + op.output_type.extend([mace_pb2.DT_FLOAT]) def convert_argmax(self, tf_op): op = self.convert_general_op(tf_op) -- GitLab