Fix (export):remove unused attribute in debug flow

Xilinx · Nov 26, 2023 · 8cfa85a · 8cfa85a
1 parent 2d4ff9b
commit 8cfa85a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/brevitas/export/onnx/debug.py b/src/brevitas/export/onnx/debug.py
@@ -51,7 +51,5 @@ def enable_debug(model, input_enabled=True, output_enabled=True, proxy_level=Fal
     for name, module in model.named_modules():
         if hasattr(module, "export_debug_name") and filter_fn(module):
             module.export_debug_name = name
-            module.export_input_debug = input_enabled
-            module.export_output_debug = output_enabled
             module.register_forward_hook(hook)
     return hook