From 716963c9947a54e148c38e61f8312ae22077fecd Mon Sep 17 00:00:00 2001 From: luxuhui Date: Tue, 7 Jan 2020 09:40:30 +0800 Subject: [PATCH] fix bug on converting LeakyRelu from ONNX N/A Signed-off-by: Luxuhui --- tools/python/transform/onnx_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/python/transform/onnx_converter.py b/tools/python/transform/onnx_converter.py index 9efaa5ee..ef4bba0e 100644 --- a/tools/python/transform/onnx_converter.py +++ b/tools/python/transform/onnx_converter.py @@ -626,7 +626,7 @@ class OnnxConverter(base_converter.ConverterInterface): else: alpha_value = 0 alpha_arg = op.arg.add() - alpha_arg.name = MaceKeyword.mace_activation_max_limit_str + alpha_arg.name = MaceKeyword.mace_activation_leakyrelu_coefficient_str alpha_arg.f = alpha_value def convert_affine(self, node): -- GitLab