From d9dbfee62343b594f3798ac32cf9e6ff592aa54e Mon Sep 17 00:00:00 2001 From: jfrery Date: Thu, 28 Mar 2024 13:12:09 +0100 Subject: [PATCH] chore: fix cifar --- .../cifar/cifar_brevitas_training/evaluate_one_example_fhe.py | 3 +-- 1 file changed, 1 insertion(+), 2 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 index fff70a89ee..afec3d6a87 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 @@ -138,9 +138,8 @@ def wrapper(*args, **kwargs): print(f"Quantization of a single input (image) took {quantization_execution_time} seconds") print(f"Size of CLEAR input is {q_x_numpy.nbytes} bytes\n") - p_error = quantized_numpy_module.fhe_circuit.p_error expected_quantized_prediction, clear_inference_time = measure_execution_time( - partial(quantized_numpy_module.fhe_circuit.simulate, p_error=p_error) + partial(quantized_numpy_module.fhe_circuit.simulate) )(q_x_numpy) # Encrypt the input