Fix the wrong index

JDAI-CV · Aug 25, 2019 · 3fe4210 · 3fe4210
1 parent ea9909e
commit 3fe4210
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/onnx2bnn/OnnxConverter.cpp b/tools/onnx2bnn/OnnxConverter.cpp
@@ -568,7 +568,7 @@ void OnnxConverter::CalculateCoeff(const ONNX_NAMESPACE::NodeProto &node,
                                                             coeff_a_data[i];
                 }
                 if (node2.input_size() == 2) {
-                    const auto &bias = onnx_float_tensors_[node2.input(2)];
+                    const auto &bias = onnx_float_tensors_[node2.input(1)];
 
                     FORZ(i, coeff_b_data.size()) {
                         coeff_b_data[i] += coeff_a_data[i] * bias.data[i];