chore: check stats without optim

zama-ai · Jun 5, 2024 · b11d99d · b11d99d
1 parent fa2a20f
commit b11d99d
Showing 1 changed file with 4 additions and 1 deletion.
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
@@ -98,7 +98,10 @@ def wrapper(*args, **kwargs):
     enable_unsafe_features=True,
     use_insecure_key_cache=True,
     insecure_key_cache_location=KEYGEN_CACHE_DIR,
-    additional_pre_processors=[tlu_optimizer, rounding],
+    additional_pre_processors=[
+        # tlu_optimizer,
+        rounding,
+    ],
     fhe_simulation=SIMULATE_ONLY,
     fhe_execution=not SIMULATE_ONLY,
 )