From 88b07d09beb4282854f2530251fc480801183229 Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Fri, 31 Mar 2023 18:17:17 +0800 Subject: [PATCH] [caffe2onnx] minor fix for Log converter MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 当然这里也只能解决 base=-1, scale=1, shift=0 这类最简单的场景 --- tools/caffe2onnx/src/OPs/Log.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/caffe2onnx/src/OPs/Log.py b/tools/caffe2onnx/src/OPs/Log.py index 327d42879..f7212c975 100644 --- a/tools/caffe2onnx/src/OPs/Log.py +++ b/tools/caffe2onnx/src/OPs/Log.py @@ -31,7 +31,7 @@ def get_log_output_shape(input_shape): def create_log_node(layer, node_name, input_name, output_name, input_shape): - output_shape = get_log_output_shape(layer) + output_shape = get_log_output_shape(input_shape) node = Node.c2oNode(layer, node_name, 'Log', input_name, output_name, input_shape, output_shape)