Remove unecessary modules attribute in debug export

Xilinx · Nov 20, 2023 · 735a96b · 735a96b
1 parent 68a6b60
commit 735a96b
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