chore: enable input compression in cifar benchmark

zama-ai · Apr 22, 2024 · d32abfd · d32abfd
1 parent ece9bdc
commit d32abfd
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.")