chore: fix torch evaluate with default rounding bitwidth

zama-ai · Apr 17, 2024 · d7b8d3c · d7b8d3c
1 parent e8d3a4d
commit d7b8d3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/use_case_examples/cifar/cifar_brevitas_training/evaluate_torch_cml.py b/use_case_examples/cifar/cifar_brevitas_training/evaluate_torch_cml.py
@@ -139,7 +139,7 @@ def main(args):
         "--rounding_threshold_bits",
         nargs="+",
         type=int,
-        default=[None],
+        default=[6],
         help="Number of bits to target with rounding.",
     )
     parser.add_argument(