chore: enable input compression in cifar benchmark

zama-ai · Apr 23, 2024 · f7b5122 · f7b5122
1 parent 702ec38
commit f7b5122
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py b/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py
@@ -82,6 +82,7 @@ def wrapper(*args, **kwargs):
     enable_unsafe_features=True,
     use_insecure_key_cache=True,
     insecure_key_cache_location=KEYGEN_CACHE_DIR,
+    compress_input_ciphertexts=True,
 )
 
 print("Compiling the model.")