Fixed bugs with Sigmoid layers in C++ ONNX parser (#733)

NeuralNetworkVerification · Feb 12, 2024 · 9818412 · 9818412
1 parent de73b92
commit 9818412
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,7 @@
 * Added support for ONNX networks with tanh nodes to command-line interface
 * Added proof producing versions of Sign, Max, Absolute Value and Disjunction constraints
 * Added support for properties provided in VNN-LIB format for ONNX networks via the Python API
+* Fixed bugs when parsing Sigmoid layers in the C++ ONNX parser.
 * Supported additional non-linear constraints Softmax and Bilinear
 * Removed dependency on torch and drop support for Python3.7
 * Bumped ONNX version to >= 1.15.0

diff --git a/src/input_parsers/NetworkParser.cpp b/src/input_parsers/NetworkParser.cpp
@@ -62,7 +62,7 @@ void NetworkParser::addRelu( Variable inputVar, Variable outputVar )
 
 void NetworkParser::addSigmoid( Variable inputVar, Variable outputVar )
 {
-    _signList.append( new SignConstraint( inputVar, outputVar ) );
+    _sigmoidList.append( new SigmoidConstraint( inputVar, outputVar ) );
 }
 
 void NetworkParser::addMaxConstraint( Variable var, Set<Variable> elements )
@@ -152,7 +152,7 @@ void NetworkParser::getMarabouQuery( InputQuery& query )
     for ( std::pair<Variable,float> upper : _upperBounds )
     {
         ASSERT( upper.first < _numVars );
-        query.setLowerBound( upper.first,upper.second );
+        query.setUpperBound( upper.first,upper.second );
     }
 }