chore: use default rounding threshold in cifar fine tuning

zama-ai · Apr 17, 2024 · 47fe458 · 47fe458
1 parent d7b8d3c
commit 47fe458
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/use_case_examples/cifar/cifar_brevitas_finetuning/cifar_utils.py b/use_case_examples/cifar/cifar_brevitas_finetuning/cifar_utils.py
@@ -459,6 +459,7 @@ def fhe_compatibility(model: Callable, data: DataLoader) -> Callable:
         torch_inputset=data,
         show_mlir=False,
         output_onnx_file="test.onnx",
+        rounding_threshold_bits=6,
     )
 
     return qmodel