Fix issue #23

Verified-Intelligence · Mar 25, 2022 · 02f6266 · 02f6266
1 parent 214710e
commit 02f6266
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_LiRPA/bound_general.py b/auto_LiRPA/bound_general.py
@@ -1179,7 +1179,7 @@ def compute_bounds(self, x=None, aux=None, C=None, method='backward', IBP=False,
 
         # check whether weights are perturbed and set nonlinear for the BoundMatMul operation
         for n in self._modules.values():
-            if isinstance(n, (BoundLinear, BoundConv, BoundBatchNormalization)):
+            if type(n) in [BoundLinear, BoundConv, BoundBatchNormalization]:
                 n.nonlinear = False
                 for l_name in n.input_name[1:]:
                     node = self._modules[l_name]