From d32abfd5d1d233c8f035758d85226b79ad158927 Mon Sep 17 00:00:00 2001 From: Roman Bredehoft Date: Mon, 22 Apr 2024 14:08:14 +0200 Subject: [PATCH] chore: enable input compression in cifar benchmark --- .../cifar/cifar_brevitas_training/evaluate_one_example_fhe.py | 1 + 1 file changed, 1 insertion(+) 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 index afec3d6a87..dbfc811f87 100644 --- 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.")