From 2f04eda83efdf735455439bb7d8eb164a2d7f3d2 Mon Sep 17 00:00:00 2001 From: lyvette Date: Mon, 24 Aug 2020 15:03:24 +0800 Subject: [PATCH] fix leaky relu parser bug. --- .../tools/converter/parser/tflite/tflite_activation_parser.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc b/mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc index 2e58536bcd3..c79caf73442 100644 --- a/mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc +++ b/mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc @@ -71,7 +71,7 @@ STATUS TfliteActivationParser::Parse(const std::unique_ptr &t return RET_NULL_PTR; } attr->alpha = tflite_attr->alpha; - attr->type = schema::ActivationType_SIGMOID; + attr->type = schema::ActivationType_LEAKY_RELU; } op->primitive->value.type = schema::PrimitiveType_Activation;