Fix (examples/ptq): fix for bitwidth check (#934)

Xilinx · Apr 16, 2024 · 6d0a8d7 · 6d0a8d7
1 parent 6de7d5a
commit 6d0a8d7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -555,7 +555,10 @@ def apply_learned_round_learning(
 def check_positive_int(*args):
     """
     We check that every inputted value is positive, and an integer.
+    If it's None, it is skipped.
     """
     for arg in args:
+        if arg is None:
+            continue
         assert arg > 0.0
         assert math.isclose(arg % 1, 0.0)